目前看到的是两题。可以用一种套路。
先看59. 螺旋矩阵 II比先看54.螺旋矩阵更容易理解。59是正方形,54是长方形。
59. 螺旋矩阵 II
参考https://leetcode.cn/problems/spiral-matrix-ii/solution/luo-xuan-ju-zhen-ii-by-leetcode-solution-f7fp/1298674
官解确实不咋地,然后一开始看的是代码随想录的,能看懂,但是写的过程需要考虑的其实有点多,变量太多,还是这个最简洁。
代码随想录给出了一个重点是 坚持循环不变量原则
,要么左闭右开,要么左开右闭,不能混着来。包括下面这个解也是。
public int[][] generateMatrix(int n) {
int t=0;
int b=n-1;
int l=0;
int r=n-1;
int k=1;
int[][] res=new int[n][n];
while (k<=n*n){
//上边
for(int i=l;i<=r;i++){
res[t][i]=k++;
}
t++;
//右边
for(int i=t;i<=b;i++){
res[i][r]=k++;
}
r--;
//下边
for(int i=r;i>=l;i--){
res[b][i]=k++;
}
b--;
//左边
for(int i=b;i>=t;i--){
res[i][l]=k++;
}
l++;
}
return res;
}
54.螺旋矩阵
参考https://leetcode.cn/problems/spiral-matrix/solution/cxiang-xi-ti-jie-by-youlookdeliciousc-3/
类似t59,区别是一个正方形,一个是长方形,正方形时最后到最里面会同时越界,但是长方形时每条边都需要判断是否越界,一旦越界需要立刻退出循环,否则其他边未越界,还会继续输出。
public static List<Integer> spiralOrder(int[][] matrix) {
int n=matrix.length;
int m=matrix[0].length;
List<Integer> res=new ArrayList<>();
int t=0;
int b=n-1;
int l=0;
int r=m-1;
while (true){
for(int i=l;i<=r;i++){
res.add(matrix[t][i]);
}
t++;
if(t>b){
break;
}
for(int i=t;i<=b;i++){
res.add(matrix[i][r]);
}
r--;
if(r<l){
break;
}
for(int i=r;i>=l;i--){
res.add(matrix[b][i]);
}
b--;
if(b<t){
break;
}
for(int i=b;i>=t;i--){
res.add(matrix[i][l]);
}
l++;
if(l>r){
break;
}
}
return res;
}