删除for中的切片元素

Question

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:]...)
    }
}

是否有比使用len或更好或更惯用的方式append？

Answer 1

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

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

因此,如果删除元素,则必须手动递减循环变量!让我们看一个例子:删除以"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]

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

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,请将结果分配给a:a = 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上的所有例子.