leetcode 54. 螺旋矩阵

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

示例 1:

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

思路

1. 四个变量设定边界
2. 四个循环。注意退出条件

code

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