【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 := 1
for k, _ := range m {
curRes := 1
// 每个key验证
if _, exist := m[k-1]; !exist {
// 第一个数，统计最长
num := k + 1
for _, ok := m[num]; ok; _, ok = m[num] {
curRes++
res = max(res, curRes)
num++
}
}
}
return res
}

正确写法2：

package main

func findContinueNum2(nums []int) int {
    numsSet := map[int]struct{}{}
    for _, val := range nums {
       numsSet[val] = struct{}{}
    }
    res := 1
    for k := range numsSet {
       curRes := 1
       if _, ok := numsSet[k-1];!ok {
          // 第一个数，起点
          num := k
          for {
             _, 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 循环使用不当的问题。

原文地址：https://blog.csdn.net/YiGeiGiaoGiao/article/details/144717235

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！