原题链接:https://leetcode.com/problems/spiral-matrix/
最初做这道题时,就是控制数组下标来实现转向,方法繁琐而且下标还容易出错。
于是我看官方解答,发现了一个更好的解法,记录学习一下。
int[] rowDirec={0,1,0,-1};
int[] colDirec={1,0,-1,0};
巧妙之处就在这两个数组,用来控制转向,初始值di=0;
而且用辅助数组seen来判断是否到撞到南墙了。
举个例子,对于M行N列的数组,最初i=0,j=0;
然后在每次迭代中都对i,j更新。
int row=i+rowDirec[di];
int col=j+colDirec[di];
由于在colDirec[0]==1时, rowDirec[0]==0,出现了i不动,j一直加一的情况,
然后只需记得在螺旋转向的同时更新di即可。
AC 1ms 100% Java:
class Solution {
public List<Integer> spiralOrder(int[][] matrix) {
List<Integer> ans=new ArrayList();
if(matrix==null||matrix.length==0)
return ans;
int m=matrix.length,n=matrix[0].length;
boolean[][] seen=new boolean[m][n];
int[] rowDirec={0,1,0,-1};
int[] colDirec={1,0,-1,0};
int i=0,j=0,di=0;
for(int count=0;count<m*n;count++){
ans.add(matrix[i][j]);
seen[i][j]=true;
int row=i+rowDirec[di];
int col=j+colDirec[di];
if(row>=0&&row<m&&col>=0&&col<n&&!seen[row][col]){
i=row;
j=col;
}else{
di=(di+1)%4;
i+=rowDirec[di];
j+=colDirec[di];
}
}
return ans;
}
}