删除 for 中的切片元素

Question

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

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:]）plus 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 上的所有示例。