题目:
给你一个正整数
n
,生成一个包含 1 到 n 2 n^2 n2 所有元素,且元素按顺时针顺序螺旋排列的 n × n n \times n n×n 正方形矩阵matrix
。
代码:
class Solution {
public:
vector<vector<int>> generateMatrix(int n) {
// dir[0], {0, 1} 向右移动
// dir[1], {1, 0} 向下移动
// dir[2], {0, -1} 向左移动
// dir[3], {-1, 0} 向上移动
int dir[4][2] = {0, 1, 1, 0, 0, -1, -1, 0}; // 方向数组
vector<vector<int>> matrix(n, vector<int>(n, 0)); // 初始化矩阵
int row = 0, col = 0; // 标记当前位置的横纵坐标
int rec = -1; // 标记当前方向
for (int k = 1; k <= n * n; k++) { // 枚举数字
matrix[row][col] = k;
if (rec != -1) { // 判断是否能够继续原有方向移动
int tx = row + dir[rec][0]; // 下个位置的横坐标
int ty = col + dir[rec][1]; // 下个位置的纵坐标
if (0 <= tx && tx < n && 0 <= ty && ty < n && matrix[tx][ty] == 0) { // 判断移动的下个位置是否合法
row = tx;
col = ty;
continue;
}
}
for (int t = 0; t < 4; t++) { // 按照右/下/上/左的顺序进行移动
int tx = row + dir[t][0];
int ty = col + dir[t][1];
if (0 <= tx && tx < n && 0 <= ty && ty < n && matrix[tx][ty]) {
row = tx;
col = ty;
rec = t; // 记录当前移动方向
break;
}
}
}
return matrix;
}
};