LeetCode994.腐烂的橘子

 看完题我觉得这不是和上一道岛屿的题一样简单嘛，然后写了将近2个小时才写出来，我的思路就是，用check()先对grid检查一下，是否有以下情况：

（如果有1的周围都是空，则这个位置用不腐烂，返回-1； 如果全是1，返回永不腐烂，返回-1；如果没有2，永不腐烂，则返回-1），

定义一个hasFresh（）方法看grid中是不是还有fresh的橘子1，然后在orangesRotting()方法算minute。

就是在while（hasFresh())中对grid全部扫描一遍，如果有2，就把它放进栈中，扫描完了就把这些2的坐标拿出来，然后把2的周围的1变成2调用turnRot()，这算是变腐烂了1次，minute++，然后又是while去检查是否还有fresh，如果还有就继续腐烂，最后直到没有了fresh就跳出循环返回minute。

但是如果是一列2，2，1，0，1，1就出现死循环了，因为第一个1腐烂后无法去腐烂后面但是又始终hasFresh，

于是我加了一个visit布尔数组，如果发现了一个2并且它没有visit才把它放进栈，然后visit改为true表示已经用它腐烂过周围了不能再次腐烂周围了，下次扫面到这个2就不会放进栈了，那么上面的情况就当第一个1变成2并且visit后就没有可以放进栈的2了，所以扫描一遍后stack还是empty，所以当扫描一遍后stack还是empty的话直接返回-1，

以下是我的代码：

class Solution {
    public int orangesRotting(int[][] grid) {
       int m = grid.length;
       int n = grid[0].length;
       int minute =0;
       boolean[][] visit = new boolean[m][n];
       if(check(grid) == -1)return -1;
       while(hasFresh(grid)){
           Stack<Integer[]> stack = new Stack<>(); 
           for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if(grid[i][j] == 2 && visit[i][j] == false){
                  visit[i][j] = true;
                  Integer[] index = new Integer[]{i,j};
                  stack.push(index);
              }
           }
         }
         if(stack.isEmpty()){
             return -1;
         }
          while(!stack.isEmpty()){
            Integer[] a = stack.pop();
            grid = turnRot(grid, a[0], a[1]);
        }
        minute++;
       }

       return minute;
      

    }

    public int[][] turnRot(int[][] grid, int i, int j){
        if(i-1>=0 && grid[i-1][j]==1)grid[i-1][j] = 2;
        if(i+1<grid.length && grid[i+1][j]==1)grid[i+1][j] = 2;
        if(j-1>=0 && grid[i][j-1]==1)grid[i][j-1] = 2;
        if(j+1<grid[0].length && grid[i][j+1]==1)grid[i][j+1] = 2;
        return grid;
    }

    public boolean hasFresh(int[][] grid){
        int m = grid.length;
       int n = grid[0].length;
       for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if(grid[i][j] == 1)return true;
           }
       }
       return false;
    }

    public int check(int[][] grid){
        int m = grid.length;
       int n = grid[0].length;
       
       //如果有1的周围都是空，则这个位置用不腐烂，返回-1
       for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if((grid[i][j] == 1)){
                   if(i-1>=0 && grid[i-1][j]!=0)continue;
                   if(i+1<grid.length && grid[i+1][j]!=0)continue;
                   if(j-1>=0 && grid[i][j-1]!=0)continue;
                   if(j+1<grid[0].length && grid[i][j+1]!=0)continue;
                   return -1;
              }
           }
       }
 
       if(statuCode == 0)return 0;
        //如果全是1，返回永不腐烂，返回-1
        statuCode = -1;
       for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if(grid[i][j] != 1){
                  statuCode =1;
              }
           }
       }
       if(statuCode == -1)return -1;
        //如果没有2，用不腐烂，则返回-1
        statuCode =-1;
      for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if(grid[i][j] == 2){
                  statuCode=1;
              }
           }
       }
       if(statuCode == -1)return -1;

       return 1;
    }

}

我这个算法写的太屎了，尤其是check方法全靠一种一种情况排除，还是看看官方题解吧，写到这里的时候，我想去看看check方法能不能优化一下，把有些情况放一起check，比如那个全是1就不用判断了，因为它包含在没有2的情况里面，但是你猜怎么了？

我发现有了visit数组后，check中的所有情况都不用check，因为他们都是使得stack为空，直接返回-1了，我只能说牛逼。所以可以删掉check方法，改成代码如下：

class Solution {
    public int orangesRotting(int[][] grid) {
       int m = grid.length;
       int n = grid[0].length;
       int minute =0;
       boolean[][] visit = new boolean[m][n];
      
       while(hasFresh(grid)){
           Stack<Integer[]> stack = new Stack<>(); 
           for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if(grid[i][j] == 2 && visit[i][j] == false){
                  visit[i][j] = true;
                  Integer[] index = new Integer[]{i,j};
                  stack.push(index);
              }
           }
         }
         if(stack.isEmpty()){
             return -1;
         }
          while(!stack.isEmpty()){
            Integer[] a = stack.pop();
            grid = turnRot(grid, a[0], a[1]);
        }
        minute++;
       }

       return minute;
      

    }

    public int[][] turnRot(int[][] grid, int i, int j){
        if(i-1>=0 && grid[i-1][j]==1)grid[i-1][j] = 2;
        if(i+1<grid.length && grid[i+1][j]==1)grid[i+1][j] = 2;
        if(j-1>=0 && grid[i][j-1]==1)grid[i][j-1] = 2;
        if(j+1<grid[0].length && grid[i][j+1]==1)grid[i][j+1] = 2;
        return grid;
    }

    public boolean hasFresh(int[][] grid){
        int m = grid.length;
       int n = grid[0].length;
       for(int i =0;i<m;i++){
           for(int j=0;j<n;j++){
              if(grid[i][j] == 1)return true;
           }
       }
       return false;
    }

}

看看官方题解的做法吧，题解用的叫多源广度优先搜索，和上一道题岛屿的数量的解法差不多，先上代码：

class Solution {
    int[] dr = new int[]{-1, 0, 1, 0};
    int[] dc = new int[]{0, -1, 0, 1};

    public int orangesRotting(int[][] grid) {
        int R = grid.length, C = grid[0].length;
        Queue<Integer> queue = new ArrayDeque<Integer>();
        Map<Integer, Integer> depth = new HashMap<Integer, Integer>();
        for (int r = 0; r < R; ++r) {
            for (int c = 0; c < C; ++c) {
                if (grid[r][c] == 2) {
                    int code = r * C + c;
                    queue.add(code);
                    depth.put(code, 0);
                }
            }
        }
        int ans = 0;
        while (!queue.isEmpty()) {
            int code = queue.remove();
            int r = code / C, c = code % C;
            for (int k = 0; k < 4; ++k) {
                int nr = r + dr[k];
                int nc = c + dc[k];
                if (0 <= nr && nr < R && 0 <= nc && nc < C && grid[nr][nc] == 1) {
                    grid[nr][nc] = 2;
                    int ncode = nr * C + nc;
                    queue.add(ncode);
                    depth.put(ncode, depth.get(code) + 1);
                    ans = depth.get(ncode);
                }
            }
        }
        for (int[] row: grid) {
            for (int v: row) {
                if (v == 1) {
                    return -1;
                }
            }
        }
        return ans;
    }
}

它每个元素用序号（行号*每行的个数+列好）来表示，用一个队列来装一层的2的序号，然后用一个Map<Integer, Integer> depth表示每个节点的深度，key是序号，value是腐烂时间，他的腐烂时间其实就是父节点的腐烂时间+1，然后遍历完一次就把队列里的2取出来反向解出行号和列号，然后把周围腐烂，把周围的序号放进队列，把所有时间，也就是父节点时间+1放入map，然后取出时间，因为每次所放入的时间都是上一次的时间+1，所以最后一次的时间就是最大时间，所以最后返回ans是没有问题的，在返回之前先检查一遍，如果还有没腐烂的橘子1就返回-1。