力扣HOT100(48)图论-腐烂的橘子

为什么必须用「多源 BFS」?

先想:如果只有一个腐烂橘子,怎么做?

这就是普通的单源 BFS:

  • 把初始腐烂橘子入队(第 0 层)
  • 每分钟处理队列里当前层的所有橘子,把它们相邻的新鲜橘子腐烂,作为下一层入队
  • 每处理完一层,时间加 1
  • 最后看有没有剩下的新鲜橘子

但题目里可能有多个初始腐烂橘子!

如果对每个腐烂橘子单独做一次 BFS,再取每个新鲜橘子的最小腐烂时间,时间复杂度会变成O(k*nm)(k 是初始腐烂橘子数),效率很低。

多源 BFS 的巧妙之处

所有初始腐烂橘子,都是 BFS 的第 0 层节点,同时入队!

  • 相当于所有腐烂橘子同时开始扩散
  • 一次遍历就能得到所有新鲜橘子的最短腐烂时间
  • 时间复杂度只有O(nm),和单源 BFS 一样

核心思路(一句话讲透)

把所有初始腐烂的橘子同时放进队列,然后像层序遍历一样,一层一层处理

  • 每一层对应一分钟
  • 处理当前层的所有腐烂橘子,把它们相邻的新鲜橘子腐烂,放进下一层
  • 最后如果还有新鲜橘子没被处理,返回-1,否则返回总层数(时间)
cpp 复制代码
class Solution {
public:
   

    int orangesRotting(vector<vector<int>>& grid) {
        int dx[4] = {-1,1,0,0};
        int dy[4] = {0,0,-1,1};

        int m = grid.size();//行数
        int n = grid[0].size();//列数
        int fresh = 0;
        int mins = 0;
        queue<pair<int,int>> q;

        for(int i = 0;i<m;i++){
            for(int j = 0;j<n;j++){
                if(grid[i][j] == 2){
                    q.push({i,j});
                }
                else if(grid[i][j] == 1){
                    fresh++;
                }
            }
        }

        while(!q.empty()&& fresh>0){
            //当q不为空且fresh新鲜数大于零
            //有新鲜的就要腐烂
            int size = q.size();//初始腐烂的个数
            //往四个方向腐烂
            for(int i = 0;i<size;i++){
                //取出队首的坐标
                int x = q.front().first;
                int y = q.front().second;
                q.pop();
                //左方:
                if(y-1>=0 && grid[x][y-1] == 1 ){
                    //腐烂
                    grid[x][y-1] = 2;
                    fresh--;
                    q.push({x,y-1});
                }

                //右方
                if(y+1<n && grid[x][y+1] == 1){
                    grid[x][y+1] = 2;
                    fresh--;
                    q.push({x,y+1});
                }

                //上方
                if(x-1>=0&&grid[x-1][y] == 1){
                    grid[x-1][y] = 2;
                    fresh--;
                    q.push({x-1,y});
                }

                //下方
                if(x+1<m && grid[x+1][y] == 1){
                    grid[x+1][y] = 2;
                    fresh--;
                    q.push({x+1,y});
                }

               
            }
             mins++;
        }
        return fresh == 0 ? mins : -1;
    }
};
相关推荐
伊玛目的门徒8 小时前
试用leetcode之典中典 二数之和问题
java·算法·leetcode
Jerry8 小时前
LeetCode 226. 翻转二叉树
算法
想做小南娘,发现自己是女生喵9 小时前
【无标题】
数据结构·算法
Kx_Triumphs11 小时前
HDU4348 To the moon(主席树区间修改模板)
算法·题解
旖-旎11 小时前
《LeetCode647 回文子串 || LeetCode 5 最长回文子串》
c++·算法·leetcode·动态规划·哈希算法
轻颂呀12 小时前
约瑟夫环问题
算法
科技大视界14 小时前
投资AI项目,传统尽调不够用了——李章虎律师拆解算法、数据、算力三大雷区
人工智能·算法·数据挖掘
郝学胜-神的一滴14 小时前
算法实战:最小k个数——大顶堆的优雅解法
开发语言·数据结构·c++·python·程序人生·算法·排序算法
Irissgwe14 小时前
算法滑动窗口
数据结构·算法