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]

示例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) {
        if (matrix.size() == 0)
            return {};

        vector<int> ret;
        int n = int(matrix.size());//行数
        int m = int(matrix[0].size());//列数
        vector<vector<int>> visit(n + 2, vector<int>(m + 2, 0));
        int i = 1, j = 1;
        visit[i][j] = 1;
        ret.push_back(matrix[0][0]);
        while (true)
        {
            int flag = 0;
            //向右走
            while (!visit[i][j + 1] && j != m)
            {
                j += 1;
                visit[i][j] = 1;
                ret.push_back(matrix[i - 1][j - 1]);
                flag = 1;
            }
            //向下走
            while (!visit[i + 1][j] && i != n)
            {
                i += 1;
                visit[i][j] = 1;
                ret.push_back(matrix[i - 1][j - 1]);
                flag = 1;
            }
            //向左走
            while (!visit[i][j - 1] && j != 1)
            {
                j -= 1;
                visit[i][j] = 1;
                ret.push_back(matrix[i - 1][j - 1]);
                flag = 1;
            }
            //向上走
            while (!visit[i - 1][j] && i != 1)
            {
                i -= 1;
                visit[i][j] = 1;
                ret.push_back(matrix[i - 1][j - 1]);
                flag = 1;
            }
            if (!flag)
                break;
        }
        return ret;
    }
};