先看代码:
package main import ( "fmt" ) func main() { var baseArr = []int{1,2,3} var part1 = baseArr[:1] // (1) var part2 = baseArr[1:] var result [][]int for i := 0; i < len(part2); i++ { tempArr := append(part1, part2[i]) // (2) result = append(result, tempArr) // (3) } fmt.Println(result) }
预期结果:[[1 2] [1 3]]
实际结果:[[1 3] [1 3]]
为什么会这个样子呢?
了解原因之前先学习一下切片和append。
关于切片,
切片包括三部分,指针+长度+容量, 切片本身不包含数据,而是由指针指向一片底层数据
这种结构的带来好处是当把切片作为函数参数时,只会拷贝切片本身,速度非常快。
但是在把切片的某部分赋给另外的切片时,修改当前切片的值,原来的切片指向的底层数组就会发生变化。
举个例子,
a := []int{1,2,3}
b := a[1:2],
b[0] = 4,
打印a会发现结果是[1,4,3]
即新的切片b只是改变了对原有切片的指针位置,但是仍然指向底层数组[]int{1,2,3}
而len(b) = 新切片结束位置 - 新切片起点位置 = 1
cap(b) = 原切片结束位置 - 新切片结束位置 = 2
关于append,
对切片append元素有两种情况考量,
如 append(slice, 1)
1)如果slice的容量还有剩余,元素1直接追加到slice指向的底层数组。
2) 如果slice没有剩余容量,append函数会创建一个新的切片,新切片容量为slice的两倍,并将slice中的数据拷贝到新切片,然后追加新元素。
有了对切片和append的这些认知,再来看文章开头提到的问题,原因在于par1指向了数组被改变了,由于par1的容量是3, part1 = [1],
循环第一次,part1 = [1,2], 底层数据仍然是[1,2,3], 这里的append会在part1的末尾追加,而part1的末尾实际指向原数组的第二个元素,即更新第二个元素,切莫以为
会追加到原数组末尾,变成[1,2,3,2]。然后append的结果赋给tempArr, 再追加到result, result中的切片数据实际仍然是baseArr指向的数据,对应的底层地址空间是相同的。
循环第二次,part1 = [1,3], 底层数据就变成了[1,3,3], 同时由于result中的数据也是共用的baseArr指向的底层数组,故而第一次循环append的数据也变成了[1,3]。
故而最后的result = [[1,3],[1,3]]
通过增加两行打印即可验证,
package main import ( "fmt" ) func main() { var baseArr = []int{1,2,3} var part1 = baseArr[:1] var part2 = baseArr[1:] var result [][]int for i := 0; i < len(part2); i++ { fmt.Println("the cap of part1:", cap(part1)) tempArr := append(part1, part2[i]) fmt.Println("the baseArr is:", baseArr) result = append(result, tempArr) } fmt.Println(result) }
输出:
the cap of part1: 3
the baseArr is: [1 2 3]
the cap of part1: 3
the baseArr is: [1 3 3]
[[1 3] [1 3]]
那么如何能得到预期的结果呢,使用make & copy函数,
package main import ( "fmt" ) func main() { var baseArr = []int{1,2,3} var part1 = make([]int, 1) copy(part1, baseArr[:1]) var part2 = baseArr[1:] var result [][]int for i := 0; i < len(part2); i++ { tempArr := append(part1, part2[i]) result = append(result, tempArr) } fmt.Println(result) }
需要注意,copy函数在执行的时候会取len(src)和len(dst)的最小长度进行复制。