Leetcode(每日一题)54. 螺旋矩阵

给你一个 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;
    }
};