螺旋矩阵（C++解法）

题目

给你一个 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)，其次向右移动到最右，此时第一行因为已经使用过了，可以将其从图中删去，体现在代码中就是重新定义上边界。判断若重新定义后，上下边界交错，表明螺旋矩阵遍历结束，跳出循环，返回答案。