leetcode 53: Spiral Matrix

最新推荐文章于 2022-07-06 10:23:15 发布

原创最新推荐文章于 2022-07-06 10:23:15 发布 · 174 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

201 篇文章

订阅专栏

本文介绍了一种螺旋遍历二维矩阵的算法实现，通过两种不同的方法来完成这一过程：一种是传统的边界递进方式，另一种则是通过判断方向并更新已访问状态的方式。这两种方法均能有效地按螺旋顺序返回矩阵中的所有元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        if(matrix.empty())
            return res;
        int m=matrix.size();
        int n=matrix[0].size();
        int time=0,start=0;
        while(time!=m*n)
        {
            for(int i=start;i<n-start;i++)
            {
                res.push_back(matrix[start][i]);
                time++;
            }
            for(int i=start+1;i<m-start;i++)
            {
                res.push_back(matrix[i][n-start-1]);
                time++;
            }
            //do next step only if the numbers are not in a horizontal line
            for(int i=n-start-2;i>=start&&m-start-1>start;i--)
            {
                res.push_back(matrix[m-start-1][i]);
                time++;
            }
            //do next step only if the numbers are not in a vertical line
            for(int i=m-start-2;i>start&&n-start-1>start;i--)
            {
                res.push_back(matrix[i][start]);
                time++;
            }
            start++;
        }
        return res;
    }
};

Updated version, imagine you go from (0,0) and change direction every time you meet a wall or some element that is visited.

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        int m=matrix.size();
        if(m==0)
            return res;
        int n=matrix[0].size();
        vector<vector<bool> > visited(m,vector<bool>(n,0));
        int i=0,j=0;
        int dir=0;
        int count=0;
        while(1)
        {
            res.push_back(matrix[i][j]);
            visited[i][j]=1;
            count++;
            if(count==m*n)
                break;
            if(dir==0)
            {
                if(j!=n-1&&visited[i][j+1]==0)
                    j++;
                else
                {
                    dir++;
                    i++;
                }
            }
            else if(dir==1)
            {
                if(i!=m-1&&visited[i+1][j]==0)
                    i++;
                else
                {
                    dir++;
                    j--;
                }
            }
            else if(dir==2)
            {
                if(j!=0&&visited[i][j-1]==0)
                    j--;
                else
                {
                    dir++;
                    i--;
                }
            }
            else if(dir==3)
            {
                if(i!=0&&visited[i-1][j]==0)
                    i--;
                else
                {
                    dir=0;
                    j++;
                }
            }
        }
        return res;
    }
};