如何在 Go 中安全地边遍历边删除链表元素

在 go 的 `container/list` 中直接边遍历边删除会导致迭代中断,因为被删节点的 `next()` 返回 `nil`;正确做法是**提前保存下一个节点指针**,再执行删除操作。

Go 标准库的 container/list 是一个双向链表实现,其 Element 结构体不持有对链表的强引用,因此在遍历时若调用 l.Remove(e),该节点即从链表中解绑——此时 e.Next() 不再指向有效后继节点(可能为 nil 或已失效),导致 for e := l.Front(); e != nil; e = e.Next() 循环提前终止。

解决的核心思路是:将 e.Next() 的值提前缓存到局部变量中,再执行可能的删除操作,最后用缓存值更新循环变量。这种“先取后删”的模式确保了迭代的连续性。

以下是修正后的去重函数完整示例:

func removeDuplicate(l *list.List) *list.List {
    seen := make(map[int]bool) // 建议使用局部变量而非全局 sMap,避免并发/复用问题
    var next *list.Element
    for e := l.Front(); e != nil; e = next {
        next = e.Next() // ✅ 关键:在任何可能修改 e 状态的操作前,先保存下一节点
        if val, ok := e.Value.(int); ok {
            fmt.Printf("VALUE: %d\n", val)
            if seen[val] {
                fmt.Printf("Deleting %d\n", val)
                l.Remove(e)
            } else {
                fmt.Printf("Adding new entry %d\n", val)
                seen[val] = true
            }
        }
    }
    return l
}

⚠️ 注意事项:

  • 永远不要在循环条件中依赖被删节点的方法调用(如 e.Next());
  • 使用 var next *list.Element 显式声明并初始化为 nil,避免未定义行为;
  • seen 映射建议定义在函数内,避免全局状态引发的竞态或副作用;
  • 类型断言 e.Value.(int) 应配合 ok 判断增强健壮性(生产环境推荐);
  • 若需保留首次出现的元素(如本例),逻辑天然成立;若需保留最后一次,则应反向遍历(Back() → Prev())并调整判断逻辑。

该模式不仅适用于去重,也适用于任意基于条件的过滤、清理或转换场景,是操作 container/list 时必须掌握的安全惯用法。