【LeetCode】螺旋矩阵

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

示例1:
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]

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

示例2:
输入:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]

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

  • 思路

定义好四个边界,按照便利的顺序依次更改边界,判断如果上下边界或者左右边界相重合的话,则证明遍历完成~

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ret;
        if(matrix.empty())
            return ret;
        int up = 0;
        int down = matrix.size();
        int left = 0;
        int right = matrix[0].size();
        while(true) {
            for(int i = left; i < right; ++i) {
                ret.push_back(matrix[up][i]);
            }
            if(++up >= down) {
                break;
            }
            for(int i = up; i < down; ++i) {
                ret.push_back(matrix[i][right - 1]);
            }
            if(--right <= left) {
                break;
            }
            for(int i = right - 1; i >= left; --i) {
                ret.push_back(matrix[down - 1][i]);
            }
            if(--down <= up) {
                break;
            }
            for(int i = down - 1; i >= up; --i) {
                ret.push_back(matrix[i][left]);
            }
            if(++left >= right) {
                break;
            }
        }
        return ret;
    }
};
发布了33 篇原创文章 · 获赞 14 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_44816732/article/details/104255271