给你一个 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]
思路:定义上下左右四条边界,然后按上、右、下、左的顺序挨个遍历,每遍历完一条边界,就立即把该边界收缩一个单位。
如图所示,上边界遍历完后(1、2、3、4),就要往下移动上边界,
然后再遍历右边界的元素8、12。
遍历完右边界,往左移动一个单位,再遍历下边界11、10、9。
然后下边界往上移动一个单位,遍历左边界,此时只有5了。
左边界往右移动一个单位,开始第二轮,遍历上边界6、7。
遍历完上边界后,要把上边界往下移动一个单位,此时上边界大于了下边界,退出循环。
所以每次遍历完一条边界后,要把该边界先向内移动一个单位,同时判断是否超过了对面的边界,如果超过了,就break,退出循环。
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
int m=matrix.size();
int n=matrix[0].size();
vector<int> ans;
int up=0,down=m-1;
int left=0,right=n-1;
while(true)
{
for(int i=left;i<=right;i++)
{
ans.push_back(matrix[up][i]);
}
if(++up>down)break;
for(int i=up;i<=down;i++)
{
ans.push_back(matrix[i][right]);
}
if(--right<left)break;
for(int i=right;i>=left;i--)
{
ans.push_back(matrix[down][i]);
}
if(--down<up)break;
for(int i=down;i>=up;i--)
{
ans.push_back(matrix[i][left]);
}
if(++left>right)break;
}
return ans;
}
};