遍历切片时去掉元素,错误示例:
package main
import ( "fmt")
func main() {
arr := []int{1, 2, 3, 4} for i := range arr { if arr[i] == 3 { // 即此时 下标为 2 println(len(arr)) //arr = append(arr[:i], arr[i+1:]...) //因为range在迭代时已经确定i的范围为[0,len(arr))的左闭右开的区间即[0,3)。 //当满足arr[i] == 3时对arr进行了修改,缩短了arr的长度,此时len(arr)=3,最大下标为2,因此当执行arr[3]时会报错。因为溢出了 }
}
fmt.Println(arr)
} //panic: runtime error: index out of range [3] with length 3那如何正确删除指定切片元素?我们稍微改下:
遍历切片时去掉元素,不会报错,但不建议的写法:
package main
import ( "fmt")
func main() {
arr := []int{1, 2, 3, 4} for i, v := range arr { if v == 3 {
arr = append(arr[:i], arr[i+1:]...) // arr[:i] 即为arr[:2]=> []int{1, 2}, arr[i+1:]即为:arr[3:] =>[]int{4} }
}
fmt.Println(arr)
} // 输出 [1 2 4]解释:
还是回到range的用法,当执行for循环时就已经确定(i,v)的遍历元素值,及时循环过程中修改了arr,也不会改变for要遍历的(i,v)值。
可以将上面代码修改一下,看下在循环中改变arr值时,后面遍历的(i,v)是不会随着arr的改变而改变的。继续往下看:
遍历切片时去掉元素,建议写法:
package main
import ( "fmt")
func main() {
arr := []int{1, 2, 3, 4} for i := 0; i < len(arr); i++ {
fmt.Println(i, arr[i]) if arr[i] == 3 {
fmt.Println("i--之前=", i)
arr = append(arr[:i], arr[i+1:]...) //arr[:2],arr[3:] i-- fmt.Println("i--之后=", i)
}
}
fmt.Println(arr)
}输出:
0 11 22 3i--之前= 2i--之后= 12 4[1 2 4]
解释:
该方案只修改i的值,在删除元素时进行i--,可以确保遍历arr没有问题,而且每次通过arr[i]获取切片值不存在问题。
当然用该方式也可以在遍历时添加元素,只要i也对应变化就没问题。
总结:
关于切片遍历时进行操作需要注意一些坑。
map遍历时进行操作相对坑少点,不过遍历map需要修改元素时,map的value要为指针类型,这点值得谨记。