给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
示例 1:
输入:n = 3 输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1 输出:[[1]]
把正方圈看作四条边加一个中心点就好办了,边取左闭右开,四条边分别以[i][j++],[i++][j],[i][j--],[i--][j]的方式遍历填入从1自增到n的值,最中间的值当n取奇数时可以直接填入而不用再走一次循环。那么,简单举例推一下n取奇数和偶数时,分别要遍历几圈(一趟四边为一圈)?
当n=3时,遍历一圈+一点;当n=4时候,遍历两圈;当n=1时,一点;当n=2时,遍历一圈。易得出以2取余的圈数,3%2=1(圈)1(点);4%2=2(圈);5%2=2(圈)1(点);6%2=3圈。因此,可以定义一个loop变量熔断循环,整个循环结束之后加个if补一下中间的洞即可。
class Solution {
public int[][] generateMatrix(int n) {
//二维的n^2数组
int [][] nums = new int [n][n];
//圈的起点,各自++后即是下(内)一圈的起点
int startX = 0;
int startY = 0;
//i为行,j为列
int i = 0;
int j = 0;
//边界,取左闭右开,n与gap的关系取几个值推演下就能理解
int gap = 1;
//填数,从1到n^2
int count = 1;
//记录圈数(循环出口)
int loop = 1;
//一圈四条边
while (loop <= n / 2) {
//从顶部→,行i不动
//列j从起点开始增加直到遍历n - gap次
for (j = startY; j < n - gap; j++) {
nums[startX][j] = count++;
}
//从右侧↓,列j不动
//行i从起点开始增加直到遍历n - gap次
for (i = startX; i < n - gap; i++) {
nums[i][j] = count++;
}
//从底部←,行i不动
//列j归还至起点
for (; j > startY; j--) {
nums[i][j] = count++;
}
//从左侧↑,列j不动
//行i归还至起点
for (; i > startX; i--) {
nums[i][j] = count++;
}
//一圈结束后需要增加厚度,改变起点,圈数膨胀(出口)
gap++;
startX++;
startY++;
loop++;
}
//奇数则补洞
if (n % 2 == 1) {
nums[startX][startY] = count;
}
//循环结束返回所求数组:nums[][]
return nums;
}
}