题目:
给你一个正整数 n
,生成一个包含 1
到 n
*n所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
示例:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
解题思路:
矩阵的填充顺序为从左上到右上,从右上到右下,从右下到左下,从左下到左上,逐层向内缩减。
我们应想办法使循环过程中遵循相同的边界条件,因此在每个方向的遍历填充过程中,遵循左闭右开的规则,右边界的元素交给下一个方向的遍历填充过程来完成,如下图所示:
每种颜色代表一个方向的填充,这样填充过程就会有规矩可遵循。
而对于向内层逐渐靠拢的过程,我们可以通过一个计数器count来记录遍历过的圈数,在每圈遍历开始前,根据count给初始点(i,j)赋值为(count,count),对于四个方向的临界点,我们得出:
从左上到右上:i=0,j<n-1-count;
从右上到右下:j=n-1,i<n-1-count;
从右下到左下:i=n-1,j>count;
从左下到左上:j=0,i>count;
像n=3,n=5这样的奇数,遍历到n*n时,会仅剩这单个元素,因此在遍历前进行判断。
代码示例:
class Solution {
public int[][] generateMatrix(int n) {
int[][] res = new int[n][n];
int start = 1, end = n * n,count = 0;
loop1:for (; start <= end; start++) {
int i = count, j = count;
if (start == end) {
res[count][count] = start;
break loop1;
}
while (j < n - 1 - count) {
res[i][j] = start;
start++;
j++;
}
while (i < n - 1 - count) {
res[i][j] = start;
start++;
i++;
}
while (j > count) {
res[i][j] = start;
start++;
j--;
}
while (i > count) {
res[i][j] = start;
start++;
i--;
}
count++;
start--;
}
return res;
}
}