如何在 Go 语言中切片的指定位置插入元素

在使用 Go 语言编程时，我们经常需要对切片（slice）进行各种操作，其中之一就是在指定位置插入元素。本文将介绍几种实现这一功能的方法，并结合具体代码示例进行说明。

使用 append 函数

最基本的方法是使用内置的 append 函数。这种方法通过两次调用 append，分别处理插入位置之前和之后的元素。

func insert(a []int, index int, value int) []int {
    if len(a) == index { // 如果在末尾插入
        return append(a, value)
    }
    a = append(a[:index+1], a[index:]...) // 扩展切片
    a[index] = value
    return a
}

func main() {
    a := []int{10, 30, 40}
    a = insert(a, 1, 20)
    fmt.Println(a) // 输出: [10 20 30 40]
}

这种方法适用于大多数情况下的插入操作，简单且易于理解。

使用标准库 slices 包中的 Insert 函数

Go 1.21 引入了 slices 包，其中提供了一个方便的 Insert 函数，可以直接用于在切片中插入元素。

import "golang.org/x/exp/slices"

func main() {
    slice := []int{1, 3, 4, 5}
    index := 1
    value := 6
    result := slices.Insert(slice, index, value)
    fmt.Println(result) // 输出: [1 6 3 4 5]
}

这个方法更简洁，并且由标准库提供，避免了手动处理边界检查和切片扩展。

自定义插入函数

有时我们需要自定义一个函数来处理更加复杂的插入逻辑，例如在不同的数据类型之间进行泛型插入。

func insert[T any](a []T, index int, value T) []T {
    n := len(a)
    if index < 0 {
        index = (index%n + n) % n
    }
    if index >= n {
        a = append(a, value)
    } else {
        a = append(a[:index+1], a[index:]...)
        a[index] = value
    }
    return a
}

func main() {
    a := []int{0, 1, 2, 3, 4}
    a = insert(a, 2, 9)
    fmt.Println(a) // 输出: [0 1 9 2 3 4]
}

这种方法使用了 Go 1.18 引入的泛型，可以适用于任何类型的切片，提高了代码的通用性。