【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].

 

 

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

转向按照右，下，左，上

 

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) {
        int m=matrix.size();
        if(m==0) return vector<int>();
       
        int n=matrix[0].size();
       
        vector<int> result(m*n);
       
        vector<vector<bool> > visited(m,vector<bool>(n,false));
       
        vector<pair<int,int>> dir(4);
        dir[0]=pair<int,int>(0,1);
        dir[1]=pair<int,int>(1,0);
        dir[2]=pair<int,int>(0,-1);
        dir[3]=pair<int,int>(-1,0);
       
        int i,j,k,count;
        i=j=k=count=0;
 
        while(1)
        {              
            if(count==m*n) break;
 
            result[count]=matrix[i][j];
            visited[i][j]=true;
 
            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])
            {
                k=(++k)%4;
            }
 
            i+=dir[k].first;
            j+=dir[k].second;
           
            count++;
        }
       
        return result;
    }
};

 

 

另外一种方法：

 

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int> > &matrix) {
        int m=matrix.size();
        if(m==0) return vector<int>();
       
        int n=matrix[0].size();
       
        vector<int> result;
       
       
        int x1=0;
        int y1=0;
        int x2=m-1;
        int y2=n-1;
       
       
        while(1)
        {
           
            for(int j=y1;j<=y2;j++) result.push_back(matrix[x1][j]);
            x1++;
           
           
            for(int i=x1;i<=x2;i++) result.push_back(matrix[i][y2]);
            y2--;
           
           
            if(x2+1!=x1)
                for(int j=y2;j>=y1;j--) result.push_back(matrix[x2][j]);
            x2--;
           
            if(y1!=y2+1)
                for(int i=x2;i>=x1;i--) result.push_back(matrix[i][y1]);
 
            y1++;
           
            if(result.size()==m*n) break;
        }
       
        return result;
    }
 
};