题目:
给你一个 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]
思路:
解题思路主要参考59. 螺旋矩阵 II 中等
依旧是坚持循环不变量原则,这里坚持左闭右闭。
终止条件是:判断若重新定义边界后,上下或者左右边界交错,表明螺旋矩阵遍历结束,跳出循环,返回答案
代码:
// 左闭右闭
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
int u = 0, d = matrix.size() - 1, l = 0, r = matrix[0].size() - 1; // 定义上下左右边界
vector<int> result; // 存储输出结果
while (1)
{
// 顺时针循环一圈依次取值并存入result中
// 存从左到右取上边一行的值
for (int i = l; i <= r; i++) result.push_back(matrix[u][i]);
if (++u > d) break; //++u更新上边界,若上边界大于下边界,则遍历完成输出
// 存从上到下取右边一列的值
for (int i = u; i <= d; i++) result.push_back(matrix[i][r]);
if (--r < l) break;
// 存从右到左取下边一行的值
for (int i = r; i >= l; i--) result.push_back(matrix[d][i]);
if (--d < u) break;
// 存从下到上取左边一列的值
for (int i = d; i >= u; i--) result.push_back(matrix[i][l]);
if (++l > r) break;
}
return result;
}
};
总结:
时间复杂度:O(mn),其中 m 和 n 分别是输入矩阵的行数和列数。矩阵中的每个元素都要被访问一次。
空间复杂度:O(mn)。需要创建一个大小为 m×n 的矩阵 visited 记录每个位置是否被访问过。
依旧是坚持循环不变量原则。
终止条件是:判断若重新定义边界后,上下或者左右边界交错,表明螺旋矩阵遍历结束,跳出循环,返回答案