从切片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
更好或更惯用的方式?
答案 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
,请将结果分配给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上的所有示例。