54. 螺旋矩阵

题目

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

输入：matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,3,6,9,8,7,4,5]

代码

定义方向数组，先右，再下，再左，再上，与bfs写法类似. 撞墙了或者走过了，就该换方向了。

class Solution2 {
private:const int dx[4] = {0, 1, 0, -1};const int dy[4] = {1, 0, -1, 0};vector<int> res;const int USED = 1e9+10;int n, m; // n rows, m columns
public:vector<int> spiralOrder(vector<vector<int>>& matrix) {n = matrix.size(); m = matrix[0].size();res.assign(m*n, 0);int d = 0; int x = 0, y = 0;for (int i = 0; i < m*n; i++) {res[i] = matrix[x][y];matrix[x][y] = USED;int a = x+dx[d]; int b = y + dy[d];if (a < 0 || a >= n || b < 0 || b >= m) // 超边界{d = (d+1) % 4;}else if (matrix[a][b] == USED) // 这里要加else，否则上面超界了还会访问matrix[a][b]{d = (d+1) % 4;}x = x + dx[d];y = y + dy[d];}return res;}
};