LintCode-374.螺旋矩阵

螺旋矩阵

给定一个包含 m x n 个要素的矩阵，（m 行, n 列），按照螺旋顺序，返回该矩阵中的所有要素。

样例

给定如下矩阵：
[
    [ 1, 2, 3 ],
    [ 4, 5, 6 ],
    [ 7, 8, 9 ]
]
应返回 [1,2,3,6,9,8,7,4,5]。

标签

数组 矩阵

code

class Solution {
public:
    /**
     * @param matrix a matrix of m x n elements
     * @return an integer array
     */
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        // Write your code here
        vector<int> order;;
        int tlx,tly,brx,bry;
        int i,j;

        if(matrix.size() == 0)
            return order;
        tlx = tly = 0;
        brx = matrix[0].size();
        bry = matrix.size();

        if(brx >= bry) {
            for(j=0; j<matrix.size()/2; j++) {
                for(i=tlx; i<brx; i++) {
                    order.push_back(matrix[tlx][i]);
                }
                for(i=tly+1; i<bry; i++)  {
                    order.push_back(matrix[i][brx-1]);
                }
                for(i=brx-2; i>=tlx; i--) {
                    order.push_back(matrix[bry-1][i]);
                }
                for(i=bry-2; i>=tly+1; i--) {
                    order.push_back(matrix[i][tlx]);
                }
                tlx++;tly++; brx--;bry--;
            }
            if(matrix.size()%2 == 1) {
                for(i=tlx; i<brx; i++) 
                    order.push_back(matrix[tlx][i]);
                }
            }
        }
        else
        {
            for(j=0; j<matrix[0].size()/2; j++) {
                for(i=tlx; i<brx; i++) {
                    order.push_back(matrix[tlx][i]);
                }
                for(i=tly+1; i<bry; i++) {
                    order.push_back(matrix[i][brx-1]);
                }
                for(i=brx-2; i>=tlx; i--) {
                    order.push_back(matrix[bry-1][i]);
                }
                for(i=bry-2; i>=tly+1; i--) {
                    order.push_back(matrix[i][tlx]);
                }
                tlx++;tly++; brx--;bry--;
            }
            if(matrix[0].size() %2 == 1) {
                for(i=tly; i<bry; i++) {
                    order.push_back(matrix[i][brx-1]);
                }
            }
        }
        return order;
    }
};