螺旋矩阵II
题目描述
给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
示例1:
输入:n = 3 输出:[[1,2,3],[8,9,4],[7,6,5]]
示例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
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 }