力扣题目:59. 螺旋矩阵 II
- 螺旋矩阵 II
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
思路
看看我之前写的博客吧,不懂的欢迎评论问我,或者你直接代码也能看懂,挺简单的。
博客链接
AC代码
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
// 二维向量这么玩~
// vector< vector<int> > matrix(n, vector<int>(n, 4)); // 二维vector,元素是向量,n个向量,每个向量是一个大小为n初值为4
vector< vector<int> > matrix(n, vector<int>(n, 0)); // 二维vector,元素是向量,n个向量,每个向量是一个大小为n初值为0
// 提前预判的思想:紫书的蛇形填数,提前预判 比 悔步 更好!
int count = 0, x = 0, y = 0; // 初始化坐标个填数个数count
y = -1; // 因为提前预判,第一个数方便用
while (count != n*n) // 当等于n*n时,填完啦
{
// 向左,提前预判, 没越界 并且 未填入(则为初始值0)
while (y + 1 < n && matrix[x][y + 1] == 0) matrix[x][ ++ y] = ++ count;
// 向下,提前预判
while (x + 1 < n && matrix[x + 1][y] == 0) matrix[ ++ x][y] = ++ count;
// 向左
while (y - 1 >= 0 && matrix[x][y - 1] == 0) matrix[x][ -- y] = ++ count;
// 向上
while (x - 1 >= 0 && matrix[x - 1][y] == 0) matrix[ -- x][y] = ++ count;
}
// for (int i = 0; i < n; i ++ )
// {
// for (int j = 0; j < n; j ++ )
// {
// cout << matrix[i][j] << " ";
// }
// cout << endl;
// }
return matrix;
}
};