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]

问题分析

此题我用的是有限状态机的方法。把当前遍历状态分为4种，分别代表不同方向。还有4个变量代表接下来可遍历的范围：行的上限和下限，列的左限和右限。每次遍历完一行或者一列后，并根据该次遍历的方向，将对应的接下来的遍历范围缩小一行或者一列，并将游标移动到下一次开始的位置，然后把遍历方向设置为接下来要遍历的方向。如此循环直到游标不在遍历范围内时退出循环。这里还记录了一个变量，记录当前还有多少个元素没有遍历，当把所有元素都遍历结束后，直接返回ans。

代码实现

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        if(matrix.empty())
            return {};
        vector<int> ans;
        string dir = "right";
        int rowup = 0;
        int colleft = 0;
        int rowdown = matrix.size() - 1;
        int colright = matrix[0].size() - 1;
        int val = (rowdown + 1) * (colright + 1);
        int i = 0;
        int j = 0;
        while(j <= colright || i <= rowdown || j >= colleft || i >= rowup){
            if(dir == "right"){
                while(j <= colright){
                    ans.push_back(matrix[i][j]);
                    j++; 
                    val--;
                    if(val == 0)
                        return ans;
                }
                j--;
                i++;
                rowup++;
                dir = "down";
            }
            else if(dir == "down"){
                while(i <= rowdown){
                    ans.push_back(matrix[i][j]);
                    i++;
                    val--;
                    if(val == 0)
                        return ans;
                }
                i--;
                j--;
                colright--;
                dir = "left";
            }
            else if(dir == "left"){
                while(j >= colleft){
                    ans.push_back(matrix[i][j]);
                    j--;
                    val--;
                    if(val == 0)
                        return ans;
                }
                j++;
                i--;
                rowdown--;
                dir = "up";
            }
            else{
                while(i >= rowup){
                    ans.push_back(matrix[i][j]);
                    i--;
                    val--;
                    if(val == 0)
                        return ans;
                }
                i++;
                j++;
                colleft++;
                dir = "right";
            }
        }
        return ans;
    }
};