LeetCode 994. 腐烂的橘子（广度优先搜索bfs）

最新推荐文章于 2024-05-20 15:46:19 发布

岁忧

最新推荐文章于 2024-05-20 15:46:19 发布

阅读量417

点赞数 9

分类专栏： LeetCode 文章标签： leetcode 宽度优先算法 c++ 广度优先搜索

本文链接：https://blog.csdn.net/weixin_46028214/article/details/138786317

版权

LeetCode 专栏收录该内容

213 篇文章 0 订阅

订阅专栏

994. 腐烂的橘子

在这里插入图片描述

思路：经典bfs广度优先搜索题。首先把腐烂的橘子加到队列qu中，同时记录还有多少个ct橘子是新鲜的。然后每次遍历当前队列中的元素（通过lens=qu.size()实现），然后把被感染的橘子也放入队列，时间加一mit++。重复上述操作，最后直到ct=0或者队列为空。

class Solution {
public:
    int orangesRotting(vector<vector<int>>& grid) {
        int n=grid.size(),m=grid[0].size();
        queue<pair<int ,int>> qu;
        vector<vector<bool>> sta(n,vector<bool>(m,false));
        int ct=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                if(grid[i][j]==1) ct++;
                else if(grid[i][j]==2){
                    qu.push({i,j});
                    sta[i][j]=1;
                }
            }
        }
        int fx[5]={0,0,1,-1},fy[5]={1,-1,0,0};
        int mit=0;
        while(qu.size()&&ct){
            int lens=qu.size();
            mit++;
            for(int k=0;k<lens;k++){
                pair<int ,int> tmp=qu.front();
                qu.pop();
                for(int i=0;i<4;i++){
                    int x=tmp.first+fx[i];
                    int y=tmp.second+fy[i];
                    if(x<0||x>=n||y<0||y>=m||sta[x][y]) continue;
                    if(grid[x][y]==1){
                        ct--;
                        qu.push({x,y});
                        sta[x][y]=1;
                    }
                }

            }
            
        }
        if(ct==0) return mit;
        else return -1;
    }
};