【LeetCode刷题记录】54. 螺旋矩阵

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]

提示：
m == matrix.length
n == matrix[i].length
1 <= m, n <= 10
-100 <= matrix[i][j] <= 100

思路

模拟矩阵路径（参考力扣评论区大神题解）
设置4个指针，分别指向行、列的开头和结尾。分4种情况讨论：
从左到右，从上到下，从右到左，从下到上，到点就退出。

代码

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> v;
        int m = matrix.size();
        int n = matrix[0].size();
        if (m == 0 || n == 0)
            return v;
        // 分别代表行的开头和结尾
        int rowS = 0, rowE = n - 1;
        // 分别代表列的开头和结尾
        int colS = 0, colE = m - 1;
        while (true) {
            // 从左到右
            for (int i = rowS; i <= rowE; i++)
                v.push_back(matrix[colS][i]);
            if (++colS > colE)
                break;

            // 从上到下
            for (int i = colS; i <= colE; i++)
                v.push_back(matrix[i][rowE]);
            if (--rowE < rowS)
                break;

            // 从右到左
            for (int i = rowE; i >= rowS; i--)
                v.push_back(matrix[colE][i]);
            if (--colE < colS)
                break;

            // 从下到上
            for (int i = colE; i >= colS; i--)
                v.push_back(matrix[i][rowS]);
            if (++rowS > rowE)
                break;
        }
        return v;
    }
};