一尘不染

删除for中的slice元素

go

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

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


阅读 246

收藏
2020-07-02

共1个答案

一尘不染

您建议的解决方案不正确。问题是,当您从切片中删除一个元素时,所有后续元素都会 移位
。但是循环并不知道您更改了基础切片,并且循环变量(索引)照常增加,即使在这种情况下也不应该这样,因为那样您就跳过了一个元素。

并且,如果切片包含2个彼此相邻的元素,则都需要删除这两个元素,则不会检查第二个元素,也不会删除它们。

因此,如果删除一个元素,则必须手动减小循环变量!我们来看一个示例:删除以开头的单词"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,则将结果分配给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上尝试所有示例。

2020-07-02