原题地址:https://leetcode-cn.com/problems/spiral-matrix-ii/description/
题目描述:
给定一个正整数 n,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的正方形矩阵。
示例:
输入: 3 输出: [ [ 1, 2, 3 ], [ 8, 9, 4 ], [ 7, 6, 5 ] ]
解题思路:
本道题思路与54题相似,只需稍微改动54题的代码即可。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> re(n, vector<int>(n, 0));
int count = 0;
int num = 1;
while((n - count*2)*(n - count*2) > 0){
int i = count, j = count;
while(j < n - count){
re[i][j++] = num ++;
}
j --;
if(++ i >= n - count) break;
while(i < n - count){
re[i++][j] = num ++;
}
i --;
if(-- j < count) break;
while(j >= count){
re[i][j--] = num ++;
}
j ++;
if(-- i <= count) break;
while(i > count){
re[i--][j] = num ++;
}
count ++;
}
return re;
}
};