题目
给定一个正整数 n,生成一个包含 1 到 n^2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
思路
注意循环不变量原则即可,不要一会左闭右开,一会左闭右闭,要始终保持相同的边界条件画下去。
然后按照上行从左到右—>右列从上到下—>下行从右到左—>左列从下到上的顺序循环走下去。
代码(java版本)
//左闭右开
class Solution {
public int[][] generateMatrix(int n) {
int[][] res = new int[n][n];
int startx = 0,starty = 0;//定义每个循环的圈的起始位置
int loop = n/2;//圈的循环次数
int mid = n/2;//矩阵最中间的位置,单独填充
int count = 1;
int offset = 1;//控制每次循环的圈的遍历长度,每次循环右边界减小一位
int i,j;
while(loop -- > 0){
i = startx;
j = starty;
//上行从左到右
for(j = starty;j < n - offset;j++){//相当于第一行的 0 ~ n-2 个格子,n-1个格子走不到,因为是右开,留给了后面
res[startx][j] = count++;
}
//右列从上到下
for(i = startx;i < n - offset;i++){//相当于最后一列的 0 ~ n-2 个格子
res[i][j] = count++;
}
//下行从右到左
for (; j > starty ; j--) {
res[i][j] = count++;
}
//左列从下到上
for(;i > startx;i--){
res[i][j] = count++;
}
//第二圈开始的时候,起始位置要各自加1, 例如:第一圈起始位置是(0, 0),第二圈起始位置是(1, 1)
startx++;
starty++;
//offset 控制每一圈里每一条边遍历的长度
offset++;
}
如果n为奇数的话,需要单独给矩阵最中间的位置赋值
if(n%2 == 1){
res[mid][mid] = count;
}
return res;
}
}