LeetCode Spiral Matrix

class Solution {

    public:

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

                    vector<int> ret;

                    int cols = 0;

                    int rows = 0;

                    if (!(rows = matrix.size()) || !(cols = matrix[0].size())) {

                            return ret;

                    }

                    int left = 0, top = 0, right = cols - 1, bottom = rows - 1;

                    while (left <= right && top <= bottom) {

                        circle_walk(left, top, right, bottom, matrix, ret);

                        left++, top++, bottom--, right--;

                    }

                    return ret;

            }



            void circle_walk(int left, int top, int right, int bottom, vector<vector<int> >& map, vector<int>& path) {

                    int rows = 0;

                    if (map.size() < 1 || map[0].size() < 1) return;

                            

                    // top row

                    for (int i=left; i<=right; i++) {

                        path.push_back(map[top][i]);

                    }



                    // right col

                    for (int i=top+1; i<=bottom; i++) {

                        path.push_back(map[i][right]);

                    }



                    // bottom row

                    for (int i=right-1; bottom != top && i>=left; i--) {

                        path.push_back(map[bottom][i]);

                    }



                    // left col

                    for (int i=bottom-1; left != right && i>top; i--) {

                        path.push_back(map[i][left]);

                    }



            }

};

又隔了好久没有写了,水一发,准备测试数据还是必要的,比如只有一行的matrix,只有一列的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) {

        vector<int> res;

        

        int m = matrix.size();

        if (m == 0) {

            return res;

        }

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

        

        int row = 0, col = -1;

        while(true) {

            for (int i=0; i<n; i++) res.push_back(matrix[row][++col]);

            if (--m == 0) break;

            for (int i=0; i<m; i++) res.push_back(matrix[++row][col]);

            if (--n == 0) break;

            for (int i=0; i<n; i++) res.push_back(matrix[row][--col]);

            if (--m == 0) break;

            for (int i=0; i<m; i++) res.push_back(matrix[--row][col]);

            if (--n == 0) break;

        }

        return res;

    }

};

 

你可能感兴趣的:(LeetCode)