[Leetcode]59.螺旋矩阵Ⅱ

时间:2022-03-30 16:51:36

给定一个正整数 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;
};