注:
要防止最后只有一行或者只有一列时的重复遍历。
题目:
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例 1:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1
输出:[[1]]
提示:
1 <= n <= 20
题解:
可以将矩阵看成若干层,首先填入矩阵最外层的元素,其次填入矩阵次外层的元素,直到填入矩阵最内层的元素。
对于每层,从左上方开始以顺时针的顺序填入所有元素。假设当前层的左上角位于(top,left),右下角位于(bottom,right),按照如下顺序填入当前层的元素。
- 从左到右填入上侧元素,依次为(top,left) 到(top,right)。
- 从上到下填入右侧元素,依次为(top+1,right) 到(bottom,right)。
- 如果left<right 且top<bottom,则从右到左填入下侧元素,依次为 (bottom,right−1) 到(bottom,left+1),以及从下到上填入左侧元素,依次为 (bottom,left) 到(top+1,left)。
填完当前层的元素之后,将left 和top 分别增加 1,将 right 和 bottom 分别减少 1,进入下一层继续填入元素,直到填完所有元素为止。
复杂度分析
时间复杂度:O(n^2),其中 n 是给定的正整数。矩阵的大小是n×n,需要填入矩阵中的每个元素。
空间复杂度:O(1)。除了返回的矩阵以外,空间复杂度是常数。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> result(n,vector<int>(n));
int top=0;
int bot=n-1;
int left=0;
int right=n-1;
int num=1;
while(top<=bot&&left<=right){
for(int i=left;i<=right;i++){
result[top][i]=num++;
}
for(int i=top+1;i<=bot;i++){
result[i][right]=num++;
}
if(top<bot){
for(int i=right-1;i>=left;i--){
result[bot][i]=num++;
}
}
if(left<right){
for(int i=bot-1;i>top;i--){
result[i][left]=num++;
}
}
top++;
bot--;
left++;
right--;
}
return result;
}
};