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
思路
模拟矩阵路径(参考力扣评论区大神题解)
设置4个指针,分别指向行、列的开头和结尾。分4种情况讨论:
从左到右,从上到下,从右到左,从下到上,到点就退出。
代码
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
vector<int> v;
int m = matrix.size();
int n = matrix[0].size();
if (m == 0 || n == 0)
return v;
// 分别代表行的开头和结尾
int colS = 0, colE = n - 1;
// 分别代表列的开头和结尾
int rowS = 0, rowE = m - 1;
while (true) {
// 从左到右
for (int i = colS; i <= colE; i++)
v.push_back(matrix[rowS][i]);
if (++rowS > rowE)
break;
// 从上到下
for (int i = rowS; i <= rowE; i++)
v.push_back(matrix[i][colE]);
if (--colE < colS)
break;
// 从右到左
for (int i = colE; i >= colS; i--)
v.push_back(matrix[rowE][i]);
if (--rowE < rowS)
break;
// 从下到上
for (int i = rowE; i >= rowS; i--)
v.push_back(matrix[i][colS]);
if (++colS > colE)
break;
}
return v;
}
};