【golang】map遍历注意事项

结论

for循环中三个表达式：
for initialization; condition; post { }
初始initialization只会评估一次，后续更新表达式值将不起作用，但post可以实现更新

• map 循环判断某个元素是否存在时常出现下面问题

for _, ok1 := numsSet[num+1]; ok1; {...更新num值，但ok1的值不会被更新，死循环
}
---------------------------------------------------------
for _, ok := m[num]; ok; _, ok = m[num] {更新num值，ok在for循环第三个表达式中被跟新，可以实现功能
}
-----------------------------------------------------------
for _, ok := m[num]; ok; {...// 内部手动更新也可以,关键点就是ok这个循环遍历只在第一次遍历的时候创建并评估_, ok = m[num]  
}

问题分析

使用for循环遍历判断某个元素是否存在，会存在一定的问题
对 for _, ok1 := numsSet[num+1]; ok1; {...} 这行代码的问题分析：

错误分析：

• 在 Go 语言中，for 语句的迭代子句可以有以下几种形式：
  • for initialization; condition; post { } // 初始initialization只会评估一次，后续更新表达式值将不起作用，但post可以更新
  • for condition { }
  • for { }

当 you 尝试使用 for _, ok1 := numsSet[num+1]; ok1; {...} 时，存在以下错误：

• 即使 you 正确接收了这两个值，如 _, ok1 := numsSet[num+1]，for 循环的条件部分 ok1 只会在循环开始时被评估一次，而不会在每次迭代后重新评估(因此想着通过函数内部修改num，再评估一次_, ok1 := numsSet[num+1]是不可行的)，因此无法实现循环的正确终止条件。

正确写法1：利用for循环第三个表达式，更新第二个判断条件
核心在于：for _, ok := m[num]; ok; _, ok = m[num] {...}

func findContinueNum2(nums []int) int {m := map[int]struct{}{}for _, val := range nums {m[val] = struct{}{}}res := 1for k, _ := range m {curRes := 1// 每个key验证if _, exist := m[k-1]; !exist {// 第一个数，统计最长num := k + 1for _, ok := m[num]; ok; _, ok = m[num] {curRes++res = max(res, curRes)num++}}}return res
}

正确写法2：

package mainfunc findContinueNum2(nums []int) int {numsSet := map[int]struct{}{}for _, val := range nums {numsSet[val] = struct{}{}}res := 1for k := range numsSet {curRes := 1if _, ok := numsSet[k-1];!ok {// 第一个数，起点num := kfor {_, ok1 := numsSet[num+1]if!ok1 {break}curRes++res = max(curRes, res)num++}}}return res
}func max(a, b int) int {if a > b {return a}return b
}

代码解释：

• 在 findContinueNum2 函数中，首先创建了一个 map[int]struct{} 来存储 nums 切片中的元素。
• 然后遍历 numsSet 中的键。
  • 当 k-1 不在 numsSet 中时，将 k 作为连续序列的起始元素。
  • 接着进入一个无限 for 循环，在每次迭代中，使用 _, ok1 := numsSet[num+1] 来检查 num+1 是否在 numsSet 中。
    • 如果 ok1 为 false，使用 break 终止内部的 for 循环。
    • 否则，更新 curRes 的值并将 num 加 1。
• max 函数用于比较 curRes 和 res 的大小，返回较大的值。

使用示例：

func main() {nums := []int{1, 2, 3, 5, 6, 7, 9, 10, 11}result := findContinueNum2(nums)println(result)
}

代码解释：

• 在 main 函数中，定义了一个整数切片 nums。
• 调用 findContinueNum2 函数并将结果存储在 result 中。
• 最后打印结果。

上述代码通过修正后的 for 循环结构，能够正确处理连续元素的查找和计数，避免了原代码中 for 循环使用不当的问题。