螺旋矩阵,比如给一个n或者一维数组,顺时针读,输出一个二维数组;或者反过来,给n平方或者m×n的一个二维数组,顺时针读取数组元素,输出一个一维数组。当然应该也有逆时针,但这里没有这个示例。
第一题
暴力解法
(参考了代码随想录那边的代码,看是看得懂,自己写就没思路……)
59. 螺旋矩阵 II
class Solution {
public int[][] generateMatrix(int n) {
// 提供n,要给出1-n²的 n*n 的二维数组
// 顺时针:右下左上,依次重复,每条边走完要更新边界值
// 初始的边界值是 上:0 右:n-1 下:n-1 左:0,起始点(0,0)
int[][] array = new int[n][n];
int loop = 0; // 控制循环次数(圈数)
int start = 0; // 每次循环的开始点(start, start)
int count = 1; // 定义填充数字
int i, j; // 横纵坐标
while (loop++ < n / 2) { // 圈数不会超过n/2圈,如n=5,两圈就行,最中间需要自己处理
// 这里采用每条边左闭右开,也就是最右边的点是作为下一条边的起点处理
// 上侧从左到右,左右是列变化,没错,另n-圈数正好是边的终点位置
for (j = start; j < n - loop; j++) array[start][j] = count++;
// 右侧从上到下
for (i = start; i < n - loop; i++) array[i][j] = count++;
// 下边从右到左
for (; j >= loop; j--) array[i][j] = count++;
// 左边从下到上
for (; i >= loop; i--) array[i][j] = count++;
start++; // 更新起始点(0,0)、(1,1)……
}
// 奇数的情况,之前的count赋给元素后++,count也对了
if (n % 2 == 1) array[start][start] = count;
return array;
}
}
时间复杂度:O(n²)
空间复杂度:O(1)
第二题
54. 螺旋矩阵
这个题乍一看是上一题的反向,但是用上一题的每条边终点留给下一条边起点就有点坑。while里四个循环不能循环完,总是会空出一行或者一列没填,再补就麻烦了。
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
int rows = matrix.length;
int cols = matrix[0].length;
List<Integer> result = new ArrayList<>();
int mloop = 0; // 行循环控制
int nloop = 0; // 列循环控制
int i, j;
while (mloop <= rows / 2 && nloop <= cols / 2) {
// 和上一题不同样了
// 从左到右 1234
for (j = nloop; j < cols - nloop; j++)
result.add(matrix[mloop][j]); // 行不变
// 从上到下 8 12
for (i = mloop + 1; i < rows - mloop; i++) // 去掉首行的终点,不作为起点了 +1
result.add(matrix[i][cols - nloop - 1]);
// 从右到左 11 10 9
if (mloop < rows) // 避免重复遍历最后一行
for (j = cols - nloop - 2; j >= nloop; j--) // 起点-1,索引-1,所以-2
result.add(matrix[rows - mloop - 1][j]); // 索引-1
// 从下到上 5 6 7
if (nloop < cols) // 避免重复遍历最后一列
for (i = rows - mloop - 2; i >= mloop; i--)
result.add(matrix[i][nloop]);
mloop++; // emm,while里++会在每次循环的循环体前执行,所以上一题也有疑问了
nloop++;
}
return result;
}
}
又是用了太长时间的一次,未完待续,有空再看(越欠越多了……)