【leetcode】Spiral Matrix

Spiral Matrix

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

[

 [ 1, 2, 3 ],

 [ 4, 5, 6 ],

 [ 7, 8, 9 ]

]

You should return [1,2,3,6,9,8,7,4,5].

 

 

如果下一步会遇到访问过的节点或者越界的节点，那么就转向。

转向按照右，下，左，上

 

 1 class Solution {

 2 public:

 3     vector<int> spiralOrder(vector<vector<int> > &matrix) {

 4         int m=matrix.size();

 5         if(m==0) return vector<int>();

 6        

 7         int n=matrix[0].size();

 8        

 9         vector<int> result(m*n);

10        

11         vector<vector<bool> > visited(m,vector<bool>(n,false));

12        

13         vector<pair<int,int>> dir(4);

14         dir[0]=pair<int,int>(0,1);

15         dir[1]=pair<int,int>(1,0);

16         dir[2]=pair<int,int>(0,-1);

17         dir[3]=pair<int,int>(-1,0);

18        

19         int i,j,k,count;

20         i=j=k=count=0;

21  

22         while(1)

23         {              

24             if(count==m*n) break;

25  

26             result[count]=matrix[i][j];

27             visited[i][j]=true;

28  

29             if(i+dir[k].first>m-1||j+dir[k].second>n-1||i+dir[k].first<0||j+dir[k].second<0||visited[i+dir[k].first][j+dir[k].second])

30             {

31                 k=(++k)%4;

32             }

33  

34             i+=dir[k].first;

35             j+=dir[k].second;

36            

37             count++;

38         }

39        

40         return result;

41     }

42 };

 

 

 

另外一种方法：

 

 1 class Solution {

 2 public:

 3     vector<int> spiralOrder(vector<vector<int> > &matrix) {

 4         int m=matrix.size();

 5         if(m==0) return vector<int>();

 6        

 7         int n=matrix[0].size();

 8        

 9         vector<int> result;

10        

11        

12         int x1=0;

13         int y1=0;

14         int x2=m-1;

15         int y2=n-1;

16        

17        

18         while(1)

19         {

20            

21             for(int j=y1;j<=y2;j++) result.push_back(matrix[x1][j]);

22             x1++;

23            

24            

25             for(int i=x1;i<=x2;i++) result.push_back(matrix[i][y2]);

26             y2--;

27            

28            

29             if(x2+1!=x1)

30                 for(int j=y2;j>=y1;j--) result.push_back(matrix[x2][j]);

31             x2--;

32            

33             if(y1!=y2+1)

34                 for(int i=x2;i>=x1;i--) result.push_back(matrix[i][y1]);

35  

36             y1++;

37            

38             if(result.size()==m*n) break;

39         }

40        

41         return result;

42     }

43  

44 };