LeetCode54-螺旋矩阵
LeetCode54-螺旋矩阵
给定一个包含 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]
一、思路
还记得那个顺时针旋转90度矩阵的题吗?
LeetCode48-旋转图像
思路一样,使用递归来解决,一层一层的保留矩阵中的数据
C++代码:
class Solution {
public:
vector<int> ans;
vector<int> spiralOrder(vector<vector<int>>& matrix) {
if (matrix.empty())
return ans;
int row_begin = 0, col_begin = 0;
int row_end = matrix.size() - 1, col_end = matrix[0].size() - 1;
int i, j;
while (row_begin < row_end && col_begin < col_end) {
i = row_begin;
for (j = col_begin; j <= col_end; j++)
ans.push_back(matrix[i][j]);
j = col_end;
for (i = row_begin + 1; i <= row_end; i++)
ans.push_back(matrix[i][j]);
i = row_end;
for (j = col_end - 1; j >= col_begin; j--)
ans.push_back(matrix[i][j]);
j = col_begin;
for (i = row_end - 1; i > row_begin; i--)
ans.push_back(matrix[i][j]);
row_begin++;
row_end--;
col_begin++;
col_end--;
}
if (row_begin == row_end) {
for (j = col_begin; j <= col_end; j++)
ans.push_back(matrix[row_begin][j]);
}
else if (col_begin == col_end) {
for (i = row_begin; i <= row_end; i++)
ans.push_back(matrix[i][col_begin]);
}
return ans;
}
};
执行效率: