Leetcode 59. 螺旋矩阵 II
题目
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix
思路
这道题只要控制好边界问题即可,遵循左闭右开区间,从左往右,从上往下,从右往左,从下往上开始遍历,循环的圈数是n / 2,如果n是奇数,那么,中间还有一个元素在循环结束之后进行填充。
代码
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n,vector<int>(n,0));// 定义一个二维数组 全部初始化为0
int startx = 0,starty = 0;// 定义每循环一个圈的起始位置
int loop = n / 2; // 每个圈循环几次,如果n时奇数3,那么loop = 1 只是循环一圈
int mid = n / 2;// 矩阵中间的位置 如果n是3, 中间的位置就是(1,1)
int count = 1;// 用来各一个矩阵中每一个空格赋值
int offset = 1;// 需要控制每一条边遍历的长度,每次循环右边界收缩一位
int i,j;
// loop 是转几圈的问题 n = 4 转两圈 n = 3 转一圈 中间元素单独赋值
while(loop --)
{
i = startx;
j = starty;
// 下面开始的四个for就是模拟转了一圈
// 模拟填充上面的行 从左到右 (左闭右开)
// 不能去掉starty 因为每一次循环至少会遍历starty这个元素
for(j = starty; j < starty + n - offset; j++)
{
// j 表示的是列 j < n - offset 让j不能遍历最后一列的元素
// count表示赋值元素
res[startx][j] = count;
count++;
}
// j这时候已经遍历到 倒数第二列元素
// 模拟从上到下填充 (左闭右开)
for(i = startx; i < startx + n - offset; i++)
{
// i表示遍历行 从上到下
res[i][j] = count;
count++;
}
// 然后开始从右往左进行遍历 进行填充元素 (左闭右开)
// j > starty 表示第一列元素不进行填充
for(; j > starty; j--)
{
res[i][j] = count++;
}
// 然后从下往上进行遍历 进行填充元素(左闭右开)
for(;i > startx; i--)
{
res[i][j] = count++;
}
// 第二圈开始的时候 那么起始位置都改变了
startx++;// 从(0,0)-> (1,1)
starty++;
// offset用于控制每一圈中每一条边遍历的长度
offset += 2;// 自己举例 测试一下
}
// 如果n是奇数 则需要单独给矩阵最中间的位置赋值
if(n % 2)
{
res[mid][mid] = count;// 中间单独的元素
}
return res;
}
};