给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
示例:
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
输入:n = 1
输出:[[1]]
提示
1 <= n <= 20
思想:一圈一圈对应位置放元素。
代码:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>>result(n,vector<int>(n,0));
int left=0;
int right=n-1;
int top=0;
int bottom=n-1;
int cur=1;
while(left<=right && top<=bottom){
for(int i=left;i<=right;i++){
result[top][i]=cur;
cur++;
}
top++;
if(top>bottom){break;}
for(int i=top;i<=bottom;i++){
result[i][right]=cur;
cur++;
}
right--;
if(left>right){break;}
for(int i=right;i>=left;i--){
result[bottom][i]=cur;
cur++;
}
bottom--;
if(top>bottom){break;}
for(int i=bottom;i>=top;i--){
result[i][left]=cur;
cur++;
}
left++;
if(left>right){break;}
}
return result;
}
};
54. 螺旋矩阵
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]
代码:
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
int m=matrix.size();
int n=matrix[0].size();
vector<int> result(m*n);
int left=0;
int right=n-1;
int top=0;
int bottom=m-1;
int k=0;
while(left<=right && top<=bottom){
for(int i=left;i<=right;i++){
result[k++]=matrix[top][i];
}
top++;
if(top>bottom){break;}
for(int i=top;i<=bottom;i++){
result[k++]=matrix[i][right];
}
right--;
if(left>right){break;}
for(int i=right;i>=left;i--){
result[k++]=matrix[bottom][i];
}
bottom--;
if(top>bottom){break;}
for(int i=bottom;i>=top;i--){
result[k++]=matrix[i][left];
}
left++;
if(left>right){break;}
}
return result;
}
};
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)