leetcode每日一题No.994腐烂的橘子

994.腐烂的橘子

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

值 0 代表空单元格；
值 1 代表新鲜橘子；
值 2 代表腐烂的橘子。
每分钟，腐烂的橘子 周围 4 个方向上相邻 的新鲜橘子都会腐烂。

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

思路

用广搜，每个腐烂的橘子一起搜，搜一轮time+1，常见用队列实现，我这里的方法没有用队列但原理相同

python实现

class Solution:
    def orangesRotting(self, grid: List[List[int]]) -> int:
        m = len(grid)
        n = len(grid[0])
        #两个集合记录腐烂和新鲜的橘子
        rotten = {(i,j) for i in range(m) for j in range(n) if grid[i][j] == 2}
        fresh = {(i,j) for i in range(m) for j in range(n) if grid[i][j] == 1}

        time = 0
        #下面只记录当前时间会腐烂的橘子，已经把腐烂扩散过的橘子拿出去
        while fresh:
            if not rotten: return -1
            rotten = {(i + di,j + dj) for i,j in rotten for di,dj in [(0,1),(0,-1),(1,0),(-1,0)] if (i + di,j+dj) in fresh}
            fresh -= rotten
            time += 1
        return time

class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        '初始化fresh水果个数fresh，腐烂水果坐标存进队列q，时间初始化为min=0'
        int min = 0, fresh = 0;
        queue<pair<int,int>> q;
        for(int i=0;i<grid.size();i++){
            for(int j=0;j<grid[0].size();j++){
                if(grid[i][j] == 1) fresh++;
                else if(grid[i][j]==2) q.push({i,j});
            }
        }


        vector<pair<int,int>> dirs = {{-1,0},{1,0},{0,-1},{0,1}};

        '腐烂橘子队列释放一位，然后扩散周围新鲜橘子，如果有扩散就给rotten标记为1
        rotten为1情况下完整扩散一次后给min+1，最后执行一次是否剩余fresh橘子的判断'
        while(!q.empty()){
            int n = q.size();
            bool rotten = false;
            for(int i =0;i<n;i++){
                auto x = q.front();
                q.pop();
                for(auto cur:dirs){
                    int i = x.first + cur.first;
                    int j = x.second + cur.second;
                    if(i>=0 && i<grid.size() && j>=0 && j<grid[0].size() && grid[i][j] == 1){
                        grid[i][j] = 2;
                        q.push({i,j});
                        fresh--;
                        rotten = true;
                        }
                }
            }
            if(rotten) min++;
        }
        return fresh? -1:min;
    }
};