1. 题目来源
链接:54. 螺旋矩阵
2. 题目解析
语法题,相当经典。蛇形矩阵。
就是 bfs
+ 方向数组。其中方向数组会在需要掉头的时候加 1 调用下一个偏移量,来改变方向。
其保持 0,1,2,3,0,1,2,3,...
这个顺序来获取增量改变方向。按照 4 的倍数来增加,所以可以记录掉头的次数,每次掉头时,有 d=(d+1)%4
来控制方向,达到 0,1,2,3
方向的循环。
- 时间复杂度: O ( n m ) O(nm) O(nm)。
- 空间复杂度: O ( n m ) O(nm) O(nm)
代码:
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
if (matrix.empty()) return {};
int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0};
int n = matrix.size(), m = matrix[0].size();
vector<vector<bool>> st(n, vector<bool>(m));
vector<int> ans;
for (int i = 0, x = 0, y = 0, d = 0; i < n * m; i ++ ) {
ans.push_back(matrix[x][y]);
st[x][y] = true;
int a = x + dx[d], b = y + dy[d];
if (a < 0 || a >= n || b < 0 || b >= m || st[a][b]) {
d = (d + 1) % 4;
a = x + dx[d], b = y + dy[d];
}
x = a, y = b;
}
return ans;
}
};