[LeetCode]螺旋矩阵
题目
代码
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
if (matrix.empty() || matrix[0].empty()) return {};
int height = matrix.size(), width = matrix[0].size();
vector<int> res;
int up = 0, down = height - 1, left = 0, right = width - 1;
while (true) {
//从左往右
for(int j=left;j<=right;j++)
res.push_back(matrix[up][j]);
//上面大于下面
if(++up>down)
break;
//从右上往右下
for(int i=up;i<=down;i++)
res.push_back(matrix[i][right]);
//右边小于左边
if(--right<left)
break;
//从右下往左下
for(int j=right;j>=left;j--)
res.push_back(matrix[down][j]);
//下边小于上面
if(--down<up)
break;
//从左下往左上
for(int i=down;i>=up;i--)
res.push_back(matrix[i][left]);
//左边超过了右边
if(++left>right)
break;
}
return res;
}
};
思路
每次遍历一行或者一列之后,需要调整边界大小(up,down,right,left四个边),并且判断是否已经遍历完成(当up>down,left>right说明要超出边界了,已经遍历完成所有)。
https://github.com/li-zheng-hao