算法训练营——day4螺旋矩阵
1 螺旋矩阵II-力扣59(中等)
1.1 题目:螺旋矩阵II
给你一个正整数 n
,生成一个包含 1
到 n2
所有元素,且元素按顺时针顺序螺旋排列的 n x n
正方形矩阵 matrix
。
示例 1:
输入:n = 3 输出:[[1,2,3],[8,9,4],[7,6,5]]
示例 2:
输入:n = 1 输出:[[1]]
提示:
1 <= n <= 20
1.2 编程模拟思路解法
首先避免起始末尾节点被重复访问,所以我们采取左开右闭 [ i , j ) 区间来进行模拟,注解中以第一圈为例,后面循环中圈内节点数减少,但是核心思路不变。
class Solution {
public int[][] generateMatrix(int n) {
int[][] ret = new int[n][n];// 二维数组存储
int startx = 0, starty = 0;// 定义起始行和列开始位置
int loop = 1;
int count = 1;// 填入的数字
int offset = 1;
// 采用左闭右开区间[i,j)
while (loop <= n / 2) {
int i = startx;
int j = starty;
// 给第一圈的第一行赋值
for (; j < n - offset; j++) {
ret[i][j] = count++;
}
// 第一圈的最后一列赋值
for (; i < n - offset; i++) {
ret[i][j] = count++;
}
// 第一圈的最后一行赋值
for (; j > starty; j--) {
ret[i][j] = count++;
}
// 第一圈第一列赋值
for (; i > startx; i--) {
ret[i][j] = count++;
}
startx++;
starty++;
offset++;
loop++;
}
if (n % 2 == 1) {
ret[startx][starty] = count;
}
return ret;
}
}