Go 语言中 append 操作与赋值操作不同,涉及修改原地址与不修改原地址的两种情况。
情况 1: 以下代码中为 LeetCode 78. 子集 的其中一种解决方案。
func subsets(nums []int) [][]int {
res := make([][]int, 0)
res = append(res, []int{})
for _, num := range nums {
sz := len(res)
for i := 0; i < sz; i++ {
// 注意:append 会直接修改 ele,应当复制后再做 append
ele := res[i]
newEle := make([]int, len(ele))
copy(newEle, ele)
res = append(res, append(newEle, num))
}
}
return res
}
若不使用数组拷贝为 newEle 的操作,直接执行 append 操作会导致 res 中的 ele 元素被修改。这是切片被原地修改的情况。
res = append(res, append(ele, num))
情况 2:append 操作与普通的切片截取操作不同,append 会返回新的引用。如下为 Leetcode 46. 全排列 的一种解决方案。若 backtest 传入的 res 为 [][]int 类型,则 append 操作后主函数 permute 中传入的 res 并不会被修改,因为实际上 res 的 append 操作仅仅只是赋予了指针变量一个新的切片引用,并不修改原值。传参添加指针后,可以通过指针直接访问地址修改原切片,达到原本的目的。
func permute(nums []int) [][]int {
output := []int{}
res := [][]int{}
for _, num := range nums {
output = append(output, num)
}
backtrack(len(nums), output, &res, 0)
return res
}
func backtrack(n int, output []int, res *[][]int, first int) {
// res = append(res, []int{1,2,3})
if first == n {
output_copy := make([]int, 0)
for _, num := range output {
output_copy = append(output_copy, num)
}
*res = append(*res, output_copy)
return
}
for i := first; i < n; i++ {
output[first], output[i] = output[i], output[first]
backtrack(n, output, res, first + 1)
output[first], output[i] = output[i], output[first]
}
}