golang copy 函数的使用

之前用 copy 不多，本以为它是个很方便的函数，没想到在做练习题时竟还是被它坑了。是我对他期望太多了。

func copy(dst, src []Type) int

基本认识：

• 它只能用于切片，不能用于 map 等任何其他类型
• 它返回结果为一个 int 型值，表示 copy 的长度

坑位一：切片 dst 需要先初始化长度

不是你定义好类型，就能将 src 完全 copy 到 dst 的，你需要初始化长度。

• 如果 dst 长度小于 src 的长度，则 copy 部分；
• 如果大于，则全部拷贝过来，只是没占满 dst 的坑位而已；
• 相等时刚好不多不少 copy 过来。

坑位二：源切片中元素类型为引用类型时，拷贝的是引用

由于只 copy 切片中的元素，所以如果切片元素的类型是引用类型，那么 copy 的也将是个引用。

如下面例子，matA 和 matB 地址不一样，但 matA[0] 和 matB[0] 的地址是一样的。

func wrongCopyMatrix() {
    matA := [][]int{
        {0, 1, 1, 0},
        {0, 1, 1, 1},
        {1, 1, 1, 0},
    }
    matB := make([][]int, len(matA))
    copy(matB, matA)
    fmt.Printf("%p, %p\n", matA, matA[0]) // 0xc0000c0000, 0xc0000c2000
    fmt.Printf("%p, %p\n", matB, matB[0]) // 0xc0000c0050, 0xc0000c2000
}

如果想 copy 多维切片中的每一个切片类型的元素，那么你需要将每个切片元素进行 初始化 并 拷贝。注意是两步：先 初始化，再 拷贝。

正确的拷贝一个多维数组：

func rightCopyMatrix() {
    matA := [][]int{
        {0, 1, 1, 0},
        {0, 1, 1, 1},
        {1, 1, 1, 0},
    }
    matB := make([][]int, len(matA))
    for i := range matA {
        matB[i] = make([]int, len(matA[i])) // 注意初始化长度
        copy(matB[i], matA[i])
    }
    fmt.Printf("%p, %p\n", matA, matA[0]) // 0xc00005c050, 0xc000018560
    fmt.Printf("%p, %p\n", matB, matB[0]) // 0xc00005c0a0, 0xc0000185c0
}