题目描述
难度 - 中等
原题链接 - 螺旋矩阵
示例1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
缩进法
解题的核心思路是按照右、下、左、上的顺序遍历数组,并使用四个变量圈定未遍历元素的边界:
随着螺旋遍历,相应的边界会收缩,直到螺旋遍历完整个数组:
只不过是反过来,让你按照螺旋的顺序生成矩阵:
代码演示:
public int[][] generateMatrix(int n) {
int[][]ans = new int[n][n];
int up = 0;
int down = n - 1;
int left = 0;
int right = n - 1;
int num = 1;
while(num <= n * n){
if(up <= down){
for(int i = left;i <= right;i++){
ans[up][i] = num++;
}
up++;
}
if(left <= right){
for(int i = up;i <= down;i++){
ans[i][right] = num++;
}
right--;
}
if(up <= down){
for(int i = right;i >= left;i--){
ans[down][i] = num++;
}
down--;
}
if(left <= right){
for(int i = down;i >= up;i--){
ans[i][left] = num++;
}
left++;
}
}
return ans;
}