【LeetCode】54. 螺旋矩阵

题目链接：https://leetcode-cn.com/problems/spiral-matrix/description/

题目描述

给定一个包含 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) {
        
        vector<int> result;
        if (matrix.empty() || matrix[0].empty()) return result;

        int m=matrix.size(),n=matrix[0].size();
        
        vector<vector<bool>> flag(m); //标记某个位置上的值是否被读取
        for (int i=0;i<m;i++)
            flag[i].resize(n);
        for (int i=0;i<m;i++)
            for (int j=0;j<n;j++)
                flag[i][j]=true;

        
        int count=0;
        int i=0,j=0;
        while(count<m*n){
            while(j<n && flag[i][j]){  //向右
                flag[i][j]=false; result.push_back(matrix[i][j++]); count++; 
            } 
            j--;i++;
            while(i<m && flag[i][j]){  //向下
                flag[i][j]=false; result.push_back(matrix[i++][j]); count++; 
            } 
            i--;j--;
            while(j>=0 && flag[i][j]){  //向左
                flag[i][j]=false; result.push_back(matrix[i][j--]); count++; 
            } 
            j++;i--;
            while(i>=0 && flag[i][j]){  //向上
                flag[i][j]=false; result.push_back(matrix[i--][j]); count++;  
            } 
            i++;j++;
        }
        
        return result;
        
    }
};