题目:
输入一个矩阵,按照从外向里以顺时针的顺序依次打印出每一个数字。
示例 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]
限制:
0 <= matrix.length <= 100
0 <= matrix[i].length <= 100
思路:
- 一圈一圈循环打印。
class Solution {
public:
vector<int> spiralOrder(vector<vector<int>>& matrix) {
if(matrix.empty() || matrix[0].size() == 0) return vector<int>();
int columns = matrix[0].size(), rows = matrix.size();
vector<int> res;
int pivot = 0; // 每个圈的起点
while(columns > pivot * 2 && rows > pivot * 2)
{
// 循环调用 loop 函数来输入。循环终止条件解释在上文中。
loop(matrix, columns, rows, pivot, res);
pivot ++;
}
return res;
}
void loop(vector<vector<int>>& matrix, int columns, int rows, int pivot, vector<int>& res)
{
// loop 函数用来向 res 数组中输入一整圈的元素
int rowEnd = rows - 1 - pivot;
int colEnd = columns - 1 - pivot;
for(int i=pivot; i<=colEnd; ++i)
{
// 第一步:从左到右打印一行
res.push_back(matrix[pivot][i]);
}
if(pivot < rowEnd)
{
for(int i=pivot+1; i<=rowEnd; ++i)
{
// 第二步:从上到下打印一列
res.push_back(matrix[i][colEnd]);
}
}
if(pivot < rowEnd && pivot < colEnd)
{
for(int i=colEnd-1; i>=pivot; --i)
{
// 第三步:从右到左打印一行
res.push_back(matrix[rowEnd][i]);
}
}
if(pivot <= rowEnd - 2 && pivot < colEnd)
{
for(int i=rowEnd-1; i>pivot; --i)
{
// 第四步:从下到上打印一列
res.push_back(matrix[i][pivot]);
}
}
}
};
参考:字节题库 - #剑29 - 简单 - 顺时针打印矩阵 - 3刷