LeetCode59. 螺旋矩阵 II

题目

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

示例

思路

规则
填充上行从左到右
填充右列从上到下
填充下行从右到左
填充左列从下到上

1. 定义上下左右四个基数
2. 每行每列填充完后行列的值需要改动
3. 如果上行填充完后，上行需要++
4. 如果右列填充完后，右列需要 - -
5. 如果下行填充完后，下行需要 - -
6. 如果左列填充完后，左列需要++

代码

package main

// generateMatrix函数用于生成一个n * n的二维矩阵，矩阵中的元素按照顺时针螺旋的顺序从1填充到n * n
func generateMatrix(n int) [][]int {
    // 创建一个二维切片slice，用于存储生成的矩阵。外层切片长度为n，每个内层切片也将被初始化为长度为n的切片，初始值都为0
    slice := make([][]int, n)
    for i := 0; i < n; i++ {
        slice[i] = make([]int, n)
    }

    // 初始化要填充到矩阵中的数字为1
    i := 1
    // 定义矩阵的左边界，初始化为0
    left, right := 0, n - 1
    // 定义矩阵的上边界，初始化为0
    up, down := 0, n - 1

    // 开始循环填充矩阵，直到填充完所有的n * n个数字
    for i <= n*n {
        // 从左到右填充矩阵的上一行
        for j := left; j <= right; j++ {
            slice[up][j] = i
            i++
        }
        // 上边界下移一行，准备填充下一行（从右上往右下填充）
        up++

        // 从上到下填充矩阵的右一列
        for j := up; j <= down; j++ {
            slice[j][right] = i
            i++
        }
        // 右边界左移一列，准备填充下一列（从右下往左下填充）
        right--

        // 从右到左填充矩阵的下一行
        for j := right; j >= left; j-- {
            slice[down][j] = i
            i++
        }
        // 下边界上移一行，准备填充下一行（从左下往左上填充）
        down--

        // 从下到上填充矩阵的左一列
        for j := down; j >= up; j-- {
            slice[j][left] = i
            i++
        }
        // 左边界右移一列，准备填充下一列（从左上往右上填充）
        left++
    }

    // 返回生成的完整矩阵
    return slice
}