给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
代码
class Solution {
static constexpr int direction[4][2]={{0,1},{1,0},{0,-1},-1,0};
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
if(matrix.size()==0 || matrix[0].size()==0) return{};
int m=matrix.size();
int n=matrix[0].size();
int total=m*n;
vector<vector<bool>>visit(m,vector<bool>(n,false));
vector<int>ans;
int a=0,b=0,directionindex=0;
for(size_t i=0;i<total;i++){
ans.emplace_back(matrix[a][b]);
visit[a][b]=true;
int aa=a+direction[directionindex][0],bb=b+direction[directionindex][1];
if(aa<0 ||aa>=m ||bb<0 ||bb>=n ||visit[aa][bb]){
directionindex=(directionindex+1)%4;
}
a+=direction[directionindex][0]; //更新(a,b)
b+=direction[directionindex][1];
}
return ans;
}
};