Leetcode #54 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].

Difficulty:Medium

简单的题。

vector<int> spiralOrder(vector<vector<int> >& matrix) {
        vector<int> ans;
        if(matrix.empty())
            return ans;
        int m = matrix.size();
        int n = matrix[0].size();
        int left = 0, up = 0;
        int right = n-1, down = m-1;
        while(right>left&&down>up)
        {
            for(int i = left;i<=right;i++)
                ans.push_back(matrix[up][i]);
            for(int i = up+1; i<down;i++)
                ans.push_back(matrix[i][right]);
            for(int i = right;i>=left;i--)
                ans.push_back(matrix[down][i]);
            for(int i = down-1; i>up;i--)
                ans.push_back(matrix[i][left]);
            left++;right--;
            up++;down--;
        }
        if(left==right&&up==down)
            ans.push_back(matrix[left][left]);
        else if(left==right&&up!=down)
            for(int i = up;i<=down;i++)
               ans.push_back(matrix[i][left]);
        else if(left!=right&&up==down)
            for(int i = left;i<=right;i++)
               ans.push_back(matrix[up][i]);
        return ans;
    }