Go 语言向函数传递数组

Go 语言向函数传递数组

在 Go 语言中，数组是值类型，因此将数组传递给函数时，将复制整个数组。如果数组非常大，这可能会导致性能问题。为了避免复制整个数组，可以通过传递切片（Slice）来传递数组的一部分，或者传递指向数组的指针。以下是向函数传递数组的示例：

1. 传递切片：

切片是对数组的引用，因此传递切片不会复制整个数组，而只是传递数组的引用和切片的长度信息。这是推荐的方式，因为它更高效且灵活。

package main

import "fmt"

// 接受一个整数切片参数
func modifySlice(s []int) {
    for i := range s {
        s[i] = s[i] * 2
    }
}

func main() {
    arr := []int{1, 2, 3, 4, 5}
    modifySlice(arr)
    fmt.Println(arr) // 输出 [2 4 6 8 10]
}

在上述示例中，modifySlice 函数接受一个整数切片参数，并将切片中的元素翻倍。原始数组 arr 在 modifySlice 函数内被修改。

2. 传递指向数组的指针：

另一种方法是将指向数组的指针传递给函数。这样，函数可以通过指针直接访问和修改数组的内容。

package main

import "fmt"

// 接受一个指向整数数组的指针参数
func modifyArray(arr *[5]int) {
    for i := range arr {
        arr[i] = arr[i] * 2
    }
}

func main() {
    arr := [5]int{1, 2, 3, 4, 5}
    modifyArray(&arr)
    fmt.Println(arr) // 输出 [2 4 6 8 10]
}

在上述示例中，modifyArray 函数接受一个指向整数数组的指针参数，并通过指针修改数组的内容。

请注意，无论使用切片还是数组指针，都可以有效地传递数组给函数，但根据需求和偏好选择合适的方法。通常来说，推荐使用切片，因为它更灵活且易于使用。