59. 螺旋矩阵 II

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

示例 1：

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

示例 2：

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

提示：

  • 1 <= n <= 20

解题思路

首先肯定是先定义一个 n x n 的二维数组 matrix，思路也比较简单，就是从左上角开始，顺时针填充数字，直到填满整个矩阵。这里需要注意的是，填充的顺序是从左到右，从上到下，从右到左，从下到上，依次循环。可以使用四个变量 left、right、top、bottom 来表示当前填充的边界，然后在每次填充完一条边后，更新边界，这样比较方便。

在编写代码的时候要注意边界值的选择和更新，避免出现数组越界的情况。

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
func generateMatrix(n int) [][]int {
	matrix := make([][]int, n)
	for i := range n {
		matrix[i] = make([]int, n)
	}

	for num, top, bottom, left, right := 1, 0, n-1, 0, n-1; num <= n*n; {
		for i := left; i <= right && num <= n*n; i++ {
			matrix[top][i] = num
			num++
		}
		top++

		for i := top; i <= bottom && num <= n*n; i++ {
			matrix[i][right] = num
			num++
		}
		right--

		for i := right; i >= left && num <= n*n; i-- {
			matrix[bottom][i] = num
			num++
		}
		bottom--

		for i := bottom; i >= top && num <= n*n; i-- {
			matrix[i][left] = num
			num++
		}
		left++
	}

	return matrix
}

总结

  • 使用四个变量 left、right、top、bottom 来表示当前填充的边界
  • 按照顺时针的顺序填充数字，从左上角开始
  • 每次填充完整的一行或一列后，更新边界，这样可以避免行列的重复填充