leetcode994. 腐烂的橘子(bfs)

在给定的网格中,每个单元格可以有以下三个值之一:

值 0 代表空单元格;
值 1 代表新鲜橘子;
值 2 代表腐烂的橘子。
每分钟,任何与腐烂的橘子(在 4 个正方向上)相邻的新鲜橘子都会腐烂。

返回直到单元格中没有新鲜橘子为止所必须经过的最小分钟数。如果不可能,返回 -1。

输入:[[2,1,1],[1,1,0],[0,1,1]]
输出:4

代码

class Solution {
    public int orangesRotting(int[][] grid) {
        Queue<int[]> queue=new LinkedList<>();
        int[][] dir=new int[][]{{-1,0},{1,0},{0,1},{0,-1}};
        for(int i=0;i<grid.length;i++)
            for(int j=0;j<grid[0].length;j++)//将初始的烂橘子入队
                if(grid[i][j]==2)
                {
                  queue.offer(new int[]{i,j});
                }
        int res=0;
        while (!queue.isEmpty())//bfs
        {
            int size=queue.size();
            for(int i=0;i<size;i++)
            {
                int[] e=queue.poll();
                int ex=e[0],ey=e[1];
                for(int[] d:dir)
                {
                    int nextX=ex+d[0],nextY=ey+d[1];
                    if(nextX<0||nextX>=grid.length||nextY<0||nextY>=grid[0].length||grid[nextX][nextY]==0||grid[nextX][nextY]==2)
                        continue;
                    grid[nextX][nextY]=2;//标记为腐烂
                    queue.offer(new int[]{nextX,nextY});
                }

            }

         res++;
        }

        int z=0;
        for(int i=0;i<grid.length;i++)
            for(int j=0;j<grid[0].length;j++)//检查是否还有没腐烂的橘子
                if(grid[i][j]==1)
                {
                   z++;
                } 
                if(z>0) return -1;//还有橘子没烂
                else if(res==0)//初始没有烂的橘子
                return 0;else 
                return res-1;
           

    }
}

猜你喜欢

转载自blog.csdn.net/weixin_44560620/article/details/107623356