Java——腐烂的橘子

题目链接

leetcode在线oj题——腐烂的橘子

题目描述

在给定的 m x n 网格 grid 中，每个单元格可以有以下三个值之一：

• 值 0 代表空单元格；
• 值 1 代表新鲜橘子；
• 值 2 代表腐烂的橘子。

每分钟，腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。

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

题目示例

示例1:

输入：grid = [[2,1,1],[1,1,0],[0,1,1]]
输出：4

示例2:
输入：grid = [[2,1,1],[0,1,1],[1,0,1]]
输出：-1
解释：左下角的橘子（第 2 行， 第 0 列）永远不会腐烂，因为腐烂只会发生在 4 个正向上。

示例3:
输入：grid = [[0,2]]
输出：0
解释：因为 0 分钟时已经没有新鲜橘子了，所以答案就是 0

题目提示

• m == grid.length
• n == grid[i].length
• 1 <= m, n <= 10
• grid[i][j] 仅为 0、1 或 2

解题思路

首先定义一个队列queue，遍历得到所有烂橘子的位置（值为2），将他们的坐标都放到队列里

然后将队列中的第一个元素取出来，将其相邻的没腐烂的橘子（值为1）的坐标放到队列里，将step++

再次重复之前的步骤：确定当前队列的长度（2），将队列的前两个元素都取出来，然后将其身边的新鲜橘子变成2，step++，如果所有位置都没有临近的新鲜橘子，那step就不++

继续重复刚才的步骤，即可得到正确答案

代码

class Solution {
    static class Pair{
        int x;
        int y;
        public Pair(int x, int y){
            this.x = x;
            this.y = y;
        }
    }
    
    public int orangesRotting(int[][] grid) {
        if(grid.length == 0 || grid == null){
            return 0;
        }
        int row = grid.length;
        int col = grid[0].length;

        Queue<Pair> queue = new LinkedList<>();
        //将所有坏掉的橘子入队
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if(grid[i][j] == 2){
                    queue.offer(new Pair(i, j));
                }
            }
        }
        
        int step = 0;
        int[][] nextPos = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
        //确定下一个位置
        while(!queue.isEmpty()){
            int size = queue.size();
            //记录当前循环是否有新的橘子变坏，有则为ture
            boolean flag = false;
            
            while(size != 0){
                Pair curPair = queue.poll();
                for (int i = 0; i < 4; i++) {
                    int nextX = curPair.x + nextPos[i][0];
                    int nextY = curPair.y + nextPos[i][1];
                    
                    if(nextX >= row || nextX < 0 || nextY >= col || nextY < 0){
                        continue;
                    }
                    
                    if(grid[nextX][nextY] == 1){
                        flag = true;
                        grid[nextX][nextY] = 2;
                        queue.offer(new Pair(nextX, nextY));
                    }
                }
                size--;
            }
            
            if(flag){
                step++;
            }
        }
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if(grid[i][j] == 1){
                    return -1;
                }
            }
        }
        return step;
    }
}