删除for

时间:2015-11-03 09:52:27

标签: go slice

从切片i中删除元素a的惯用方法,保留顺序,似乎是:

a = append(a[:i], a[i+1:]...)

我想知道在循环中哪个是最好的方法。据我了解,不可能在以下范围内使用它:

for i := range a { // BAD
    if conditionMeets(a[i]) {
        a = append(a[:i], a[i+1:]...)
    }
}

但是可以使用len(a)。 [编辑:这不起作用,见下面的答案]

for i := 0; i < len(a); i++ {
    if conditionMeets(a[i]) {
        a = append(a[:i], a[i+1:]...)
    }
}

是否有比使用lenappend更好或更惯用的方式?

1 个答案:

答案 0 :(得分:10)

您提出的解决方案不正确。问题是当您从切片中删除元素时,所有后续元素都将移位。但循环不知道你改变了底层切片和循环变量(索引)像往常一样递增,即使在这种情况下它不应该因为你跳过一个元素。

如果切片包含两个彼此相邻的元素,这两个元素都需要被删除,第二个元素将不会被检查,也不会被删除。

因此,如果删除元素,则必须手动递减循环变量!让我们看一个例子:删除以"a"开头的单词:

func conditionMeets(s string) bool {
    return strings.HasPrefix(s, "a")
}

解决方案(尝试使用下面Go Playground上的所有其他示例):

a := []string{"abc", "bbc", "aaa", "aoi", "ccc"}
for i := 0; i < len(a); i++ {
    if conditionMeets(a[i]) {
        a = append(a[:i], a[i+1:]...)
        i--
    }
}
fmt.Println(a)

输出:

[bbc ccc]

或者更好:使用向下循环,因此您不需要手动递减变量,因为在这种情况下,移位的元素位于切片的“已处理”部分。 / p>

a := []string{"abc", "bbc", "aaa", "aoi", "ccc"}
for i := len(a) - 1; i >= 0; i-- {
    if conditionMeets(a[i]) {
        a = append(a[:i], a[i+1:]...)
    }
}
fmt.Println(a)

输出相同。

替换多次删除

如果你必须删除“很多”元素,这可能会很慢,因为你必须进行大量的复制(append()复制)。想象一下:你有一个包含1000个元素的切片;只需删除第一个元素就需要将999个元素复制到前面。此外,还会创建许多新的切片描述符:每次删除元素都会创建 2 个新切片描述符(a[:i]a[i+1:]加上 a必须更新(append()的结果)。在这种情况下,将不可移除元素复制到新切片可能更有效。

有效的解决方案:

a := []string{"abc", "bbc", "aaa", "aoi", "ccc"}
b := make([]string, len(a))
copied := 0
for _, s := range(a) {
    if !conditionMeets(s) {
        b[copied] = s
        copied++
    }
}
b = b[:copied]
fmt.Println(b)

此解决方案分配与源长度相同的切片,因此不会执行新的分配(和复制)。此解决方案还可以使用range循环。如果您想要结果a,请将结果分配给aa = b[:copied]

输出相同。

许多删除(以及一般用途)的就地备用

我们还可以通过维护2个索引并在同一切片中分配(复制转发)不可移除元素来“循环”删除。

要记住的一件事是我们应该删除已删除元素的位置,以便删除无法访问的值的引用,以便GC可以完成其工作。这也适用于其他解决方案,但仅在此处提及。

示例实施:

a := []string{"abc", "bbc", "aaa", "aoi", "ccc"}
copied := 0
for i := 0; i < len(a); i++ {
    if !conditionMeets(a[i]) {
        a[copied] = a[i]
        copied++
    }
}
for i := copied; i < len(a); i++ {
    a[i] = "" // Zero places of removed elements (allow gc to do its job)
}
a = a[:copied]
fmt.Println(a)

输出是一样的。尝试Go Playground上的所有示例。