54. Spiral Matrix(螺旋数组)

原题链接: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;
    }
}

猜你喜欢

转载自blog.csdn.net/God_Mood/article/details/88060892
今日推荐