Leetcode第54题(螺旋矩阵)

题目描述:

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

示例 1：
输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

示例 2：
输入：matrix = [[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) {
        //m 行   n 列
        int m = matrix.size(), n = matrix[0].size();
        //上下左右四个边界
        int up = 0, down = m - 1,left = 0, right = n - 1;
        vector<int> ans;
        while(true){
            //向右走
            for(int i = left; i <= right; i++) ans.push_back(matrix[up][i]);
            if(++up > down) break;
            //向下走
            for(int i = up; i <= down; i++) ans.push_back(matrix[i][right]);
            if(--right < left) break;
            //向左走
            for(int i = right; i >= left;i--) ans.push_back(matrix[down][i]);
            if(--down < up) break;
            //向上走
            for(int i = down; i >= up; i--) ans.push_back(matrix[i][left]);
            if(++left > right) break;
        }
        return ans;
    }
};