54.螺旋矩阵 (Medium)*

题目描述*

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

示例*

输入:

[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

输出: [1,2,3,6,9,8,7,4,5]

代码*

设置四个边界，遍历同时调整边界，当边界交叉时结束。

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> res;
        if(matrix.size() == 0 || matrix[0].size() == 0) {
            return res;
        }
        int up = 0, bottom = matrix.size() - 1, left = 0, right = matrix[0].size() - 1;
        while(true) {
            for(int i = left; i <= right; i++) {
                res.push_back(matrix[up][i]);
            }
            if(++up > bottom) {
                break;
            }
            for(int i = up; i <= bottom; i++) {
                res.push_back(matrix[i][right]);
            }
            if(--right < left) {
                break;
            }
            for(int i = right; i >= left; i--) {
                res.push_back(matrix[bottom][i]);
            }
            if(--bottom < up) {
                break;
            }
            for(int i = bottom; i >= up; i--) {
                res.push_back(matrix[i][left]);
            }
            if(++left > right) {
                break;
            }
        }
        return res;
    }
};

最后更新: July 23, 2022