LeetCode 0059.螺旋矩阵II

螺旋矩阵II

题目描述

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

示例1：

1
2


输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]

示例2：

1
2


输入：n = 1
输出：[[1]]

思路

题目要求

输入一个正整数n，要求生成一个二维矩阵，元素按顺时针螺旋顺序排列。
返回生成的二维矩阵

模拟顺时针画矩阵的过程:

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

由外向内一圈一圈这么画下去。

生成一个 n×n 空矩阵 matrix，随后模拟整个向内环绕的填入过程：

定义左右上下边界 left=0,right=n-1,top=0,bottom=n-1，初始值 num = 1，迭代终止值 target = n * n；
当 num <= tar 时，始终按照 从左到右 从上到下 从右到左 从下到上 填入顺序循环，每次填入后：
- 执行 num += 1：得到下一个需要填入的数字；
- 更新边界
  - 从左到右填完后，上边界 top += 1，相当于上边界向内缩 1
  - 从上到下填完后，右边界 right -= 1，相当于右边界向内缩 1
  - 从右到左填完后，下边界 bottom -= 1，相当于下边界向内缩 1。
  - 从下到上填完后，左边界 left += 1，相当于左边界向内缩 1。
最终返回 matrix 即可。

注意

填入一个数字后记得num+1
填入一行/列后记得更新边界

代码

Go

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36


func generateMatrix(n int) [][]int {
    left, right, top, bottom := 0, n-1, 0, n-1
    num := 1
    target := n * n
    matrix := make([][]int, n)
    for i := 0; i < n; i++ {
        matrix[i] = make([]int, n)
    }
    for num <= target {
        // left to right
        for i := left; i <= right; i++ {
            matrix[top][i] = num
            num++
        }
        top++
        // top to bottom
        for i := top; i <= bottom; i++ {
            matrix[i][right] = num
            num++
        }
        right--
        // right to left
        for i := right; i >= left; i-- {
            matrix[bottom][i] = num
            num++
        }
        bottom--
        // bottom to top
        for i := bottom; i >= top; i-- {
            matrix[i][left] = num
            num++
        }
        left++
    }
    return matrix
}

Link

GitHub