59 螺旋矩阵II
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目描述
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3
输出:
[
[ 1, 2, 3 ],
[ 8, 9, 4 ],
[ 7, 6, 5 ]
]
题目给出的接口为:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
}
};
题目分析
题目很简单,我们可以认为模拟一下这个过程即可。
代码如下:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int> >ans(n,vector<int>(n));
int sum = 0 , u = 0 , d = n - 1 , l = 0 , r = n - 1;
while(sum < n * n)
{
for(int i = l ; i <= r ; i ++)
{
sum ++;
ans[u][i] = sum;
}
u++;
for(int i = u ; i <= d ; i ++)
{
sum ++;
ans[i][r] = sum;
}
r--;
for(int i = r ; i >= l ; i --)
{
sum ++;
ans[d][i] = sum;
}
d--;
for(int i = d ; i >= u ; i --)
{
sum ++;
ans[i][l] = sum;
}
l++;
}
return ans;
}