leetcode链接:https://leetcode.cn/problems/spiral-matrix-ii/description/
给你一个正整数 n ,生成一个包含 1 到 n2 所有元素,且元素按顺时针顺序螺旋排列的 n x n 正方形矩阵 matrix 。
输入:n = 3
输出:[[1,2,3],[8,9,4],[7,6,5]]
本题并不涉及到什么算法,就是模拟过程,但却十分考察对代码的掌控能力
本题依然是要坚持循环不变量原则:模拟顺时针画矩阵的过程:
- 填充上行从左到右;
- 填充右列从上到下;
- 填充下列从右到左;
- 填充左列从下到上。
由外向内一圈一圈这么画下去。
这里一圈下来,我们要画没四条边,这四条边怎么画,没画一条边都要坚持一致的左闭右开的原则,这样这一圈才能按照统一的规则画下来。
代码:
class Solution {
public int[][] generateMatrix(int n) {
int loop = 0; // 控制循环次数
int[][] res = new int[n][n];
int start = 0; // 每次循环的开始点(start, start)
int count = 1; // 定义填充数字
int i, j;
while (loop++ < n / 2) { // 判断边界后,loop
// 模拟上侧从左到右
for (j = start; j < n - loop; j++) {
res[start][j] = count++;
}
// 模拟右侧从上到下
for (i = start; i < n - loop; i++) {
res[i][j] = count++;
}
// 模拟下侧从右到左
for (; j >= loop; j--) {
res[i][j] = count++;
}
// 模拟左侧从下到上
for (; i >= loop; i--) {
res[i][j] = count++;
}
start++;
}
if (n % 2 == 1) {
res[start][start] = count;
}
return res;
}
}