LeetCode - 54. Spiral Matrix

最新推荐文章于 2023-06-20 07:29:52 发布

weixin_34260991

最新推荐文章于 2023-06-20 07:29:52 发布

阅读量101

点赞数

CC 4.0 BY-SA版权

原文链接：https://yq.aliyun.com/articles/441250

本文介绍了一种螺旋矩阵遍历算法，该算法可以顺时针漩涡状打印二维数组的值。通过控制四个边界实现遍历，时间复杂度为O(N)。文章还提供了完整的C++实现代码。

54. Spiral Matrix

Problem's Link

----------------------------------------------------------------------------

Mean:

顺时针漩涡状打印二维数组的值.

analyse:

控制四个边界即可.

Time complexity: O(N)

view code

#include <bits/stdc++.h>
using namespace std;

class Solution
{
public:
    vector<int> spiralOrder(vector<vector<int>> & matrix)
    {
        int row=matrix.size();
        if(!row) return vector<int>(0);
        int col=matrix[0].size();
        if(!col) return vector<int>(0);
        vector<int> res(row*col);
        int u=0,d=row-1,l=0,r=col-1,cnt=0;
        while(1)
        {
            for(int i=l;i<=r;++i)
                res[cnt++]=matrix[u][i];
            ++u;
            if(u>d) break;

            for(int i=u;i<=d;++i)
                res[cnt++]=matrix[i][r];
            --r;
            if(l>r) break;

            for(int i=r;i>=l;--i)
                res[cnt++]=matrix[d][i];
            --d;
            if(u>d) break;

            for(int i=d;i>=u;--i)
                res[cnt++]=matrix[i][l];
            ++l;
            if(l>r) break;
        }
        return res;
    }
};

int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        vector<vector<int>> mat(n,vector<int>(m,0));
        for(int i=0;i<n;++i)
        {
            for(int j=0;j<m;++j)
                cin>>mat[i][j];
        }
        Solution solution;
        vector<int> res=solution.spiralOrder(mat);
        for(auto p:res)
            cout<<p<<" ";
        cout<<endl;
    }
    return 0;
}
/*

*/