golang 数组转置

1 2 3
4 5 6
7 8 9

1 4 7
2 5 8
3 6 9

var array [n]Type

var a [5]int // 定义一个长度为5的int数组
a[0] = 1     // 给第0个元素赋值为1

var array [m][n]Type

var matrix = [3][3]int{
    {1, 2, 3},
    {4, 5, 6},
    {7, 8, 9},
}

func transpose(a [][]int, m, n int) [][]int {
    // 创建一个新的n*m的数组
    b := make([][]int, n)
    for i := 0; i < n; i++ {
        b[i] = make([]int, m)
    }

    // 将a的行与b的列对应
    for i := 0; i < n; i++ {
        for j := 0; j < m; j++ {
            b[i][j] = a[j][i]
        }
    }

    return b
}

func main() {
    matrix := [3][3]int{
        {1, 2, 3},
        {4, 5, 6},
        {7, 8, 9},
    }

    // 转置前
    fmt.Println(matrix)

    // 转置后
    transposedMatrix := transpose(matrix[:], 3, 3)
    fmt.Println(transposedMatrix)
}

[[1 2 3] [4 5 6] [7 8 9]]
[[1 4 7] [2 5 8] [3 6 9]]