题目
给你一个 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]
C++代码
#include <iostream>
#include <vector>
using namespace std;
/*
* 首先设定上下左右边界(left,right,top,bottom)
* 其次向右移动到最右,此时第一行因为已经使用过了,可以将其从图中删去,体现在代码中就是重新定义上边界
* 判断若重新定义后,上下边界交错,表明螺旋矩阵遍历结束,跳出循环,返回答案
*/
vector<int> spiralOrder(vector<vector<int>>& matrix) {
vector<int> ans;
if (matrix.empty()) {
return ans;
}
int left = 0, right = matrix[0].size() - 1, top = 0, bottom = matrix.size() - 1;
while (true) {
for (int i = left; i <= right; ++i) ans.push_back(matrix[top][i]);
if (++top > bottom) break;
for (int i = top; i <= bottom; ++i) ans.push_back(matrix[i][right]);
if (--right < left) break;
for (int i = right; i >= left; --i) ans.push_back(matrix[bottom][i]);
if (--bottom < top) break;
for (int i = bottom; i >= top; --i) ans.push_back(matrix[i][left]);
if (++left > right) break;
}
return ans;
}
int main() {
vector<vector<int>> matrix = { {1,2,3},{4,5,6},{7,8,9} };
vector<int> ans = spiralOrder(matrix);
for (int i = 0; i < ans.size(); i++) {
cout << ans[i] << " ";
}
cout << endl;
return 0;
}
分析
首先设定上下左右边界 (left, right, top, bottom),其次向右移动到最右,此时第一行因为已经使用过了,可以将其从图中删去,体现在代码中就是重新定义上边界。判断若重新定义后,上下边界交错,表明螺旋矩阵遍历结束,跳出循环,返回答案。