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
题解:
- 我们
按层遍历
即可,从矩阵的最外面一层一直向里扩展,有点类似于“撕包菜”
,从最外层一层一层的撕。这样模拟的话逻辑比较清晰,我们只需要对每一层进行四个方向的遍历即可。 - 具体的,我们使用四个变量来保存此时的层数对应矩阵的最下方
bottom
,最上方top
,最左方left
,最右方right
,接着就是进行模拟操作,模拟顺时针遍历的操作,一圈模拟完全后,需要对四个临界进行更新。
注意的是由于我们四个边界已经确定,所以一旦在模拟的过程中
num
已经大于n*n
了,其由于边界的限制也不会使得n*n
以外的值添加到数组中去。即一旦num
大于n*n
了,那么此时的left
,right
等就已经不合乎常理了。
依此结论,我们也可以得出外层控制层数的循环的判断条件可以写两种,第一种是基于num
和n * n
的关系,另一种是基于四个边界的大小关系。
代码:
class Solution {
public int[][] generateMatrix(int n) {
int num = 1;
int left = 0;
int right = n-1;
int top = 0;
int bottom = n-1;
int[][] matrix = new int[n][n];
while(left<=right && top<=bottom){ //这里也可写while(num<=n*n){
for(int column=left;column<=right;column++){
matrix[top][column] = num;
num++;
}
for(int row=top+1;row<=bottom;row++){
matrix[row][right] = num;
num++;
}
for(int column=right-1;column>left;column--){
matrix[bottom][column] = num;
num++;
}
for(int row=bottom;row>top;row--){
matrix[row][left] = num;
num++;
}
left++;
right--;
top++;
bottom--;
}
return matrix;
}
}