LeetCode - 54. Spiral Matrix

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;
}
/*

*/