题目:
链接:剑指 Offer 29. 顺时针打印矩阵;LeetCode 54. 螺旋矩阵
难度:中等
给你一个 m 行 n 列的矩阵 matrix ,请按照 顺时针螺旋顺序 ,返回矩阵中的所有元素。
示例 1:
输入:matrix = [[1,2,3],[4,5,6],[7,8,9]]
输出:[1,2,3,6,9,8,7,4,5]
示例 2:
输入:matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出:[1,2,3,4,8,12,11,10,9,5,6,7]
提示:
- m == matrix.length
- n == matrix[i].length
- 1 <= m, n <= 10
- -100 <= matrix[i][j] <= 100
模拟:
可以模拟打印矩阵的路径。初始位置是矩阵的左上角,初始方向是向右,当路径超出界限或者进入之前访问过的位置时,顺时针旋转,进入下一个方向。
判断路径是否进入之前访问过的位置需要使用一个与输入矩阵大小相同的辅助矩阵 visited,其中的每个元素表示该位置是否被访问过。当一个元素被访问时,将 visited 中的对应位置的元素设为已访问。
如何判断路径是否结束?由于矩阵中的每个元素都被访问一次,因此路径的长度即为矩阵中的元素数量,当路径的长度达到矩阵中的元素数量时即为完整路径,将该路径返回。
代码:
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
vector<int> ans;
if(matrix.empty()) return ans;
int n = matrix.size(), m = matrix[0].size();
vector<vector<int>> directions{{0,1}, {1,0}, {0,-1}, {-1,0}}; // 顺时针的四个方向
vector<vector<bool>> visited(n, vector<bool>(m, false)); // 访问标志位
int num = n * m; // 矩阵数字总数
int i = 0, j = 0; // 遍历坐标
int directionIndex = 0; // 方向标志
while(num--)
{
visited[i][j] = true;
ans.emplace_back(matrix[i][j]);
int ii = i + directions[directionIndex][0];
int jj = j + directions[directionIndex][1];
if(ii >= n || ii < 0 || jj >= m || jj < 0 || visited[ii][jj]) { //检查是否越界和是否被访问过
directionIndex = (directionIndex + 1) % 4; // 换方向
i += directions[directionIndex][0];
j += directions[directionIndex][1];
}
else {
i = ii;
j = jj;
}
}
return ans;
}
};
时间复杂度O(MN)。
空间复杂度O(MN)。