给你一个正整数 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
思路分析
1.定义循环几圈
2.定义二维数组
3.定义每次循环起始位置
4.给矩阵中的空格赋值
5.定义 i, j (行列)
6.左闭右开,遵循循环不变量原则, 判断边界后,loop从1开始
7.填充上行从左到右
8.填充右列从上到下
9.填充下行从右到左
10.填充左列从下到上
11.如果n为奇数的话,需要单独给矩阵最中间的位置赋值
class Solution {
public int[][] generateMatrix(int n) {
int loop = 0; //控制循环次数
int[][] res = new int [n][n];
int start = 0; //每次循环的开始节点
int count = 1; //定义填充数字
int i, j;
while(loop++ < n/2){//判断边界后,loop从1开始
//模拟上侧从左到右
for(j = start; j < n -loop; j++){
res[start][j] = count++;
}
//模拟右侧从上到下
for(i = start; i < n - loop; i++){
res[i][j] = count++;
}
//模拟左侧从上到下
for(; j >= loop; j--){
res[i][j] = count++;
}
//模拟左侧从上到下
for(; i >= loop; i--){
res[i][j] = count++;
}
// 第二圈开始的时候,起始位置要各自加1, 例如:第一圈起始位置是(0, 0),第二圈起始位置是(1, 1)
start++;
}
// 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
if(n % 2 == 1){
res[start][start] = count;
}
return res;
}
}