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