【LeetCode】54. Spiral Matrix

题目描述

Given a matrix of m x n elements (m rows, n columns), return all elements of the matrix in spiral order.

For example,
Given the following matrix:

[
 [ 1, 2, 3 ],
 [ 4, 5, 6 ],
 [ 7, 8, 9 ]
]

You should return [1,2,3,6,9,8,7,4,5].

解题思路

方向+长度。
由于需要做螺旋，所以考虑水平和竖直方向移动次数的变化规律
通过记录一个移动的方向，每次安装移动方向更新前进，值得最终需要的移动次数为0，即可。
假定一个虚拟的入口，作为移动的开始位置。

AC代码

class Solution {
public:
    vector<int> spiralOrder(vector<vector<int>>& matrix) {
        vector<int> ans;
        int nRow = matrix.size();
        if (nRow == 0) return ans;
        int nCol = matrix[0].size();
        if (nCol == 0) return ans;

        // left, down, right, up
        vector<vector<int>> dirs = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
        int nSteps[2] = {nCol, nRow - 1};
        int curDir = 0;
        int curRow = 0, curCol = -1;
        int cnt = 0;
        //使用curDir%2来判定当前是水平还是竖直方向，用来确定当前需移动长度
        while(nSteps[curDir%2]) {
            // move forward and get number
            curRow += dirs[curDir][0];
            curCol += dirs[curDir][1];
            ans.push_back(matrix[curRow][curCol]);
            cnt += 1;

            // 更新需移动的长度和方向
            if (cnt == nSteps[curDir%2]) {
                cnt = 0;
                nSteps[curDir%2]--;
                curDir = (curDir + 1) % 4;
            }
        }

        return ans;
    }
};