给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
Solution:
蛇形环绕,为了减少判断或者循环的代码,我们需要环绕一圈不变的变量作为参照量。
于是!我们设置一个变量i
,这个i
的意思表示第i
外层。
一圈的填数如下:
从左到右,从i行i列
->i行n-i-1列
从上到下,从i+1行n-i-1列
->n-i-1行n-i-1列
从右到左,从n-i-1行n-i-2列
->n-i-1行i列
从下到上,从n-i-2行i列
->i+1行i列
同时我们设置一个变量count
,用于计数保证循环结束。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> ans(n,vector<int>(n));
int count = 1,i=0;
while(count<=n*n){
int j = i;
while(j<n-i)
ans[i][j++] = count++;
j = i+1;
while(j<n-i)
ans[j++][n - i-1] = count++;
j = n - i-2;
while(j>i)
ans[n -i-1][j--] = count++;
j = n-i-1;
while(j>=i+1)
ans[j--][i] = count++;
i++;
}
return ans;
};