lavos/cap3/ex20.go

package cap3

// Ex20 趣味矩阵
func Ex20(n int) [][]int {
    matrix := make([][]int, n)
    for i := 0; i < len(matrix); i++ {
        matrix[i] = make([]int, n)
    }

    for i := range n {
        for j := range n {
            // 对角线
            if i == j || i+j == n-1 {
                matrix[i][j] = 0
                continue
            }

            // 左上部分
            if i+j < n-1 {
                if i < j {
                    // 上
                    matrix[i][j] = 1
                } else {
                    // 左
                    matrix[i][j] = 2
                }
            }

            // 右下边部分
            if i+j > n-1 {
                if i > j {
                    // 下
                    matrix[i][j] = 3
                } else {
                    // 右
                    matrix[i][j] = 4
                }
            }
        }
    }

    return matrix
}