【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;    }};