LeetCode-59. 螺旋矩阵 II

给你一个正整数 n ，生成一个包含 1 到 n2 所有元素，且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。

 

示例 1：

输入：n = 3
输出：[[1,2,3],[8,9,4],[7,6,5]]
示例 2：

输入：n = 1
输出：[[1]]
 

提示：

1 <= n <= 20

 

思路与上题一致：

#include <iostream>
#include <vector>
using namespace std;
class Solution {

public:
	vector<vector<int>> generateMatrix(int n) {
		vector<vector<int>> arr(n, vector<int>(n, 0));
		vector<vector<bool>> isWalk(n, vector<bool>(n, false));
		constructNum(arr, isWalk, 0, 0, 1);
		return arr;
	}

	void constructNum(vector<vector<int>>& arr,
		vector<vector<bool>>&isWalk,
		int x, int y, int num) {
		while (y < arr[x].size() && isWalk[x][y] == false) {
			arr[x][y] = num++;
			isWalk[x][y] = true;
			y++;
		}
		x++;
		y--;
		while (x < arr.size() && isWalk[x][y] == false) {
			arr[x][y] = num++;
			isWalk[x][y] = true;
			x++;
		}
		x--;
		y--;
		while (y >= 0 && isWalk[x][y] == false) {
			arr[x][y] = num++;
			isWalk[x][y] = true;
			y--;
		}
		x--;
		y++;
		while (x >= 0 && isWalk[x][y] == false) {
			arr[x][y] = num++;
			isWalk[x][y] = true;
			x--;
		}
		x++;
		y++;

		if (isWalk[x][y] == false) {
			constructNum(arr, isWalk, x, y, num);
		}
		return;
	}

private:
	vector<vector<bool>> isWalk;

};