给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
代码:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
int loop=n/2;
vector<vector<int>> result(n, vector<int>(n, 0));
int count=1;
int startrow,startcolumn=0;
int offset=1;
int i,j;
while(loop--){
int row=startrow;
int column=startcolumn;
for(i=column;i<n-offset;i++){
result[row][i]=count++; //从左到右,i列到最大值
}
for(j=row;j<n-offset;j++){
result[j][i]=count++;//从上到下,j行到最大值
}
for(;i>column;i--){
result[j][i]=count++;//从右到左,i列到最小值
}
for(;j>row;j--){
result[j][i]=count++;//从下到上,j行到最小值
}
startrow++;
startcolumn++;
offset++;
}
if(n%2){
result[(n/2)][(n/2)]=count;
}
return result;
}
};
可以想成从外圈往里圈写数,总共有n/2圈。每一圈的四边都分别用for循环写入数值。每条边写数时注意最后一个的数值是在下一条边开始时写入的,每一圈结束后需要缩进一圈。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/spiral-matrix-ii