go append切片之坑

先看代码：

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)的最小长度进行复制。

猜你喜欢