题目来源:https://leetcode.cn/problems/spiral-matrix-ii/description/
![](https://i-blog.csdnimg.cn/blog_migrate/6679107f7f408f73d88a24f8b873fb6a.png)
C++题解:先使用vector初始化一个二维数组,使其值都为0。由于数组的长宽都是n,所以用一个for循环,从1到n*n加入到二维数组;在for循环中只需要判断每次在哪转弯。
转弯的规律为:r+0,c+1;r+1,c+0;r+0,c-1;r-1,c+0。
知道规律就好办了,接下来只需要判断拐角。向下转弯的拐角为matrix[r][c+1]不等于0,向左转弯的拐角为matrix[r+1][c]不等于0,向上转弯的拐角为matrix[r][c-1]不等于0,向右转弯的拐角为matrix[r-1][c]不等于0,即遇到墙壁即使拐弯。同时要防止matrix溢出。代码如下:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int> > a(n, vector(n,0));
int len = n * n;
int r = 0, c = 0; //行列索引
int v = 0, h = 1; //行列增减,初始为向右走 01 01 10 10 0-1 0-1 -10 01
for(int ii = 0; ii < len; ii++){
a[r][c] = ii + 1;
if(v == 0 && h == 1 && (c == n-1 || a[r][c+1] != 0)){
v = 1; h = 0;
}
else if(v == 1 && h ==0 && (r == n-1 || a[r+1][c] != 0)){
v = 0; h = -1;
}
else if(v == 0 && h == -1 && (c == 0 || a[r][c-1] != 0)){
v = -1; h = 0;
}
else if(v == -1 && h == 0 && (a[r-1][c] != 0)){
v = 0; h = 1;
}
r = r + v; // 水平方向的移动
c = c + h; // 垂直方向的移动
}
return a;
}
};
C++题解2(来源网络):这个的思路是把螺旋矩阵看成一个一个圈,然后再把中心点单拎出来。
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
vector<vector<int>> res(n, vector<int>(n, 0)); // 使用vector定义一个二维数组
int startx = 0, starty = 0; // 定义每循环一个圈的起始位置
int loop = n / 2; // 每个圈循环几次,例如n为奇数3,那么loop = 1 只是循环一圈,矩阵中间的值需要单独处理
int mid = n / 2; // 矩阵中间的位置,例如:n为3, 中间的位置就是(1,1),n为5,中间位置为(2, 2)
int count = 1; // 用来给矩阵中每一个空格赋值
int offset = 1; // 需要控制每一条边遍历的长度,每次循环右边界收缩一位
int i,j;
while (loop --) {
i = startx;
j = starty;
// 下面开始的四个for就是模拟转了一圈
// 模拟填充上行从左到右(左闭右开)
for (j = starty; j < n - offset; j++) {
res[startx][j] = count++;
}
// 模拟填充右列从上到下(左闭右开)
for (i = startx; i < n - offset; i++) {
res[i][j] = count++;
}
// 模拟填充下行从右到左(左闭右开)
for (; j > starty; j--) {
res[i][j] = count++;
}
// 模拟填充左列从下到上(左闭右开)
for (; i > startx; i--) {
res[i][j] = count++;
}
// 第二圈开始的时候,起始位置要各自加1, 例如:第一圈起始位置是(0, 0),第二圈起始位置是(1, 1)
startx++;
starty++;
// offset 控制每一圈里每一条边遍历的长度
offset += 1;
}
// 如果n为奇数的话,需要单独给矩阵最中间的位置赋值
if (n % 2) {
res[mid][mid] = count;
}
return res;
}
(其实我觉得我的方法比较好,只是判断拐弯的时候要细心一点)