力扣刷题打卡day10 螺旋矩阵II59

题目描述

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

思路一：顺时针遍历

• 因为规定了顺时针方向所以可以按照“右-下-左-上”的方向依次循环
• 注意判断边界以及转换方向即可
• 时间复杂度为$O(n^2)$
• 空间复杂度为$O(n^2)$

class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
	vector<vector<int>>newMat;
	char dir='r';
	int row=0,col=0,cnt=1;
	for(int i=0;i<n;i++){
		vector<int> New;
		newMat.push_back(New);
		for(int j=0;j<n;j++){
			newMat[i].push_back(0);
		}
	}
	while(true){
		switch (dir){
			case 'r':
				while(row<n&&row>=0&&col<n&&col>=0&&newMat[row][col]==0){
					newMat[row][col]=cnt;
					col++;
					cnt++;
				}
				col--;
				dir='d';
				row++;
				break;
			case 'l':
				while(row<n&&row>=0&&col<n&&col>=0&&newMat[row][col]==0){
					newMat[row][col]=cnt;
					col--;
					cnt++;
				}
				col++;
				dir='u';
				row--;
				break;
			case 'd':
				while(row<n&&row>=0&&col<n&&col>=0&&newMat[row][col]==0){
					newMat[row][col]=cnt;
					row++;
					cnt++;
				}
				row--;
				dir='l';
				col--;
				break;
			case 'u':
				while(row<n&&row>=0&&col<n&&col>=0&&newMat[row][col]==0){
					newMat[row][col]=cnt;
					row--;
					cnt++;
				}
				row++;
				dir='r';
				col++;
				break;
			default:
				break;
			}
			if(cnt>n*n){
				return newMat;
				break;
			}
		}
		return newMat;
	}
};