题目
给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。
示例 1:
输入:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
输出: [1,2,3,6,9,8,7,4,5]
示例 2:
输入:
[
[1, 2, 3, 4],
[5, 6, 7, 8],
[9,10,11,12]
]
输出: [1,2,3,4,8,12,11,10,9,5,6,7]
思路
控制四个边界循环打印
实现
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
vector<int> res;
if (matrix.empty()) return res;
//初始化边界
int u = 0;
int d = matrix.size() - 1;
int l = 0;
int r = matrix[0].size() - 1;
while (1) {
for (int i = l; i <= r; i++) {
res.push_back(matrix[u][i]);
}
//更新上边界
if (++u > d) break;
for (int i = u; i <= d; i++) {
res.push_back(matrix[i][r]);
}
//更新右边界
if (--r < l) break;
for (int i = r; i >= l; i--) {
res.push_back(matrix[d][i]);
}
//更新下边界
if (--d < u) break;
for (int i = d; i >=u; i--) {
res.push_back(matrix[i][l]);
}
//更新左边界
if (++l > r) break;
}
return res;
}
};