螺旋矩阵

问题描述

给定一个包含 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]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/spiral-matrix

思路

变量说明:
vector ret 为最后返回的结果,u为螺旋矩阵按照顺时针旋转时开始的上边界,d为下边界,l为左边界,r为右边界

按照顺时针旋转的顺序,向右:i从l到r,ret.push_back(matrixu),重新定义上边界,u++,如果u>d,跳出循环;向下:i从u到d, ret.push_back(matrixi),重新定义右边界,r--,如果r>l,跳出循环;向左:i从r到l,ret.push_back(matrixd),重新定义下边界,d--,如果u>d,跳出循环;向上:i从d到u,ret.push_back(matrixi),重新定义左边界,l++,如果r>l,跳出循环。

题解

class Solution {,
public:
    vector<int> spiralOrder(vector<vector<int>> &matrix) {
        vector<int> ret;
        int n = matrix.size();
        if (n == 0) return ret;
        int u, d, l, r;
        u = 0;
        l = 0;
        d = n - 1;
        r = matrix[0].size() - 1;
        while (true) {
            for (int i = l; i <= r; i++) {
                ret.push_back(matrix[u][i]);
            }
            u++;
            if(u>d) break;
            for(int i=u;i<=d;i++){
                ret.push_back(matrix[i][r]);
            }
            r--;
            if(r<l) break;
            for(int i=r;i>=l;i--){
                ret.push_back(matrix[d][i]);
            }
            d--;
            if(d<u) break;
            for(int i=d;i>=u;i--){
                ret.push_back(matrix[i][l]);
            }
            l++;
            if(l>r) break;
        }
        return  ret;
    }
};

标签: none

添加新评论