Question

我是新手语言。我希望在根据go语言中的条件迭代列表时从列表中删除元素。例如，我想从列表中删除重复的元素。代码如下。

package main
import (
    "container/list"
    "fmt"
)
var sMap map[int]bool
func main() {
    l := list.New()
    l.PushFront(4)
    l.PushFront(5)
    l.PushFront(7)
    l.PushFront(6)
    l.PushFront(5)
    l.PushFront(4)
    l.PushFront(5)
    l.PushFront(7)
    l.PushBack(9)
    l = removeDuplicate(l)
    for e := l.Front(); e != nil; e = e.Next() {
        fmt.Println(e.Value)
    }
}
func removeDuplicate(l *list.List) *list.List {
    sMap = make(map[int]bool)
    for e := l.Front(); e != nil; e = e.Next() {
        m := e.Value.(int)
        fmt.Println("VALUE : ", m)
        if sMap[m] == true {
            fmt.Println("Deleting ", e.Value)
            l.Remove(e)
        } else {
            fmt.Println("Adding New Entry", e.Value)
            sMap[m] = true
        }
    }
    return l
}

上面的代码只遍历列表，直到第一次删除。我试图在迭代相同的列表时删除元素。这就是它无法正常工作的原因。任何人都可以在golang中建议列表迭代器吗？

Answer 1

如果从列表中删除e，则在下一个循环中调用e.Next()将返回nil。因此，在删除e.Next()之前，需要将next分配给e。以下是通过迭代（在list_test.go）

中清除所有元素的示例

// Clear all elements by iterating
var next *Element
for e := l.Front(); e != nil; e = next {
    next = e.Next()
    l.Remove(e)
}

可以将相同的模式应用于以下问题;

package main
import (
    "container/list"
    "fmt"
)
var sMap map[int]bool
func main() {
    l := list.New()
    l.PushFront(4)
    l.PushFront(5)
    l.PushFront(7)
    l.PushFront(6)
    l.PushFront(5)
    l.PushFront(4)
    l.PushFront(5)
    l.PushFront(7)
    l.PushBack(9)
    l = removeDuplicate(l)
    for e := l.Front(); e != nil; e = e.Next() {
        fmt.Println(e.Value)
    }
}
func removeDuplicate(l *list.List) *list.List {
    sMap = make(map[int]bool)
    var next *list.Element
    for e := l.Front(); e != nil; e = next {
        m := e.Value.(int)
        next = e.Next()
        fmt.Println("VALUE : ", m)
        if sMap[m] == true {
            fmt.Println("Deleting ", e.Value)
            l.Remove(e)
        } else {
            fmt.Println("Adding New Entry", e.Value)
            sMap[m] = true
        }
    }
    return l
}

输出

VALUE :  7
Adding New Entry 7
VALUE :  5
Adding New Entry 5
VALUE :  4
Adding New Entry 4
VALUE :  5
Deleting  5
VALUE :  6
Adding New Entry 6
VALUE :  7
Deleting  7
VALUE :  5
Deleting  5
VALUE :  4
Deleting  4
VALUE :  9
Adding New Entry 9
7
5
4
6
9

如何在golang中迭代相同的列表时从列表中删除元素

1 个答案: