golangcopy函数使用的坑

2023-05-14 09:05:24 函数 golangcopy

之前用 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
}

到此这篇关于golang copy函数使用的坑的文章就介绍到这了,更多相关Golang copy函数内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持！

相关文章