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
思路
从左到右,从上到下,从右到左,从下到上依次生成数字,直到边界溢出
代码
class Solution {
public int[][] generateMatrix(int n) {
int[][] result = new int[n][n];
int left=0,right=n-1;
int up=0,down=n-1;
int num=1;
while(true){
for(int i=left;i<=right;i++) result[up][i]=num++;
if(++up > down) break;
for(int i=up;i<=down;i++) result[i][right]=num++;
if(--right < left) break;
for(int i=right;i>=left;i--) result[down][i]=num++;
if(--down < up) break;
for(int i=down;i>=up;i--) result[i][left]=num++;
if(++left > right) break;
}
return result;
}
}