题目描述
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; }};