スパイラル マトリックス (C++ ソリューション)

トピック

m 行と列 n の行列が 与えられた場合 matrix 、  行列内のすべての要素を時計回りのスパイラル順に返してください。

例 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]

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;
}

分析する

まず上下左右の境界線（左、右、上、下）を設定し、次に右に移動します このとき、1行目はすでに使用されているため、画像から削除しても問題ありません。上限を再定義することでコードに反映されます。判定が再定義されると、上下の境界が交錯し、スパイラル行列の走査が完了し、ループが抜け出して、答えが返されることを示します。