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
通过次数
342.6K
提交次数
473.2K
通过率
72.4%

class Solution {
public:	
	enum direct {
		Right,
		Down,
		Left,
		Up,
	};

	vector<vector<int>> generateMatrix(int n) {
		
		vector<vector<int>> vec(n, vector<int>(n, 0));
		if (n < 0) {
			return vec;
		}
		vector<vector<bool>> flag(n, vector<bool>(n, false));
		int idx = 1;
		int x = 0, y = 0;
		direct d = Right;
		int count = n * n;
		while (count > 0) {
			vec[x][y] = idx++;
			flag[x][y] = true;
			count--;
			if (d == Right) {
				if (y + 1 < n && !flag[x][y+1]) {
					y++;
				}
				else {
					x++;
					d = Down;
				}
			}
			else if (d == Down) {
				if (x + 1 < n && !flag[x + 1][y]) {
					x++;
				}
				else {
					y--;
					d = Left;
				}
			}
			else if (d == Left) {
				if (y - 1 > -1 && !flag[x][y-1]) {
					y--;
				}
				else {
					x--;
					d = Up;
				}
			}
			else if (d == Up) {
				if (x - 1 > 0 && !flag[x - 1][y]) {
					x--;
				}
				else {
					y++;
					d = Right;
				}
			}

		}
		return vec;
	}
}