go语言list循环删除元素(转载)

本篇系转载转载地址: http://ju.outofmemory.cn/entry/79972

在使用go的container/list的package时,你可能会无意间踩一个小坑,那就是list的循环删除元素。

list删除元素,直观写下来的代码如下:

复制代码
package main

import (
    "container/list"
    "fmt"
)

func main() {
    //初始化一个list
    l := list.New()
    l.PushBack(1)
    l.PushBack(2)
    l.PushBack(3)
    l.PushBack(4)

    fmt.Println("Before Removing...")
    //遍历list,删除元素
    for e := l.Front(); e != nil; e = e.Next() {
        fmt.Println("removing", e.Value)
        l.Remove(e)
    }
    fmt.Println("After Removing...")
    //遍历删除完元素后的list
    for e := l.Front(); e != nil; e = e.Next() {
        fmt.Println(e.Value)
    }
}
复制代码

以上代码很简单,按常理来看,应该能得到正确的结果,list最后将会被清空。可事实却完全不是这样,执行后结果如下:

Before Removing...
removing 1
After Removing...
2
3
4

从结果可以看出,list根本没有清空,而只是删除了第一个元素。这是为何?原因就在container/list package的实现上了。这应该是我见过的实现最简单list了,出去注释也就100来行实现代码,而且它不只是一个简单链表,而且可以当做stack,当做queue来使用。

下面是Remove方法的代码:

复制代码
// remove removes e from its list, decrements l.len, and returns e.
func (l *List) remove(e *Element) *Element {
  e.prev.next = e.next
  e.next.prev = e.prev
  e.next = nil // avoid memory leaks
  e.prev = nil // avoid memory leaks
  e.list = nil
  l.len--
  return e
}
复制代码

这下问题原因就明显了,就出现在e.next = nil 这行代码上。当执行玩remove,e.next就变成了nil,list遍历当然也就终止了。找出问题的原因,我们就容易找到workaround的办法了,将e.next用中间变量保存起来就OK了,代码如下:

复制代码
package main

import (
  "container/list"
  "fmt"
)

func main() {
  l := list.New()
  l.PushBack(1)
  l.PushBack(2)
  l.PushBack(3)
  l.PushBack(4)
  fmt.Println("Before Removing...")
  var n *list.Element
  for e := l.Front(); e != nil; e = n {
      fmt.Println("removing", e.Value)
      n = e.Next()
      l.Remove(e)
  }
  fmt.Println("After Removing...")
  for e := l.Front(); e != nil; e = e.Next() {
      fmt.Println(e.Value)
  }
}
复制代码

 执行结果如下:

Before Removing...
removing 1
removing 2
removing 3
removing 4
After Removing..
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值