多源广度优先搜索

我们知道用队列实现广度优先搜索有三个步骤：

• 初始化队列
• 将起始点入队列
• 循环：当队列不为空时，先弹出首个元素，然后将这个元素能够处理的其他点全部入队列

而多源广度优先搜索就是起始点入队列时，我们让多个点进入队列。重点在于：如何记录广度优先搜索的最短时间，如下题：

P994腐烂的橘子

初始我们将腐烂的橘子都入队列，并记录新鲜的橘子的个数，如果最后还剩余新鲜的橘子我们就返回-1。如何处理这一时刻遍历的腐烂的橘子的最小时间呢，因为我们是让所有最初腐烂的橘子同一时刻开始传递腐烂，因此我们可以借助动态规划的思想：此时刻遍历的橘子的被腐烂的时间，一定等于让他腐烂的橘子的腐烂时间+1。，因此我们可以在原数组上直接修改，假设此时是grid[x][y]遍历到grid[dx][dy]，那么我们可以得出：grid[dx][dy]=grid[x][y]+1；最后我们考虑最初的橘子的值是2，但是他的腐烂时间是0，因此此时刻此橘子的腐烂时间ans=grid[dx][dy]-2。代码如下：

int orangesRotting(vector<vector<int>>& grid) {
    queue<pair<int,int>>q;
    int cnt=0;
    int n=grid.size();
    int m=grid[0].size();
    for(int i=0;i<n;i++){ //对数组做预处理
        for(int j=0;j<m;j++){
            if(grid[i][j]==2){
                q.push(make_pair(i,j)); //最初腐烂的橘子入队列
            }
            else if(grid[i][j]==1) cnt++; //记录新鲜的橘子的个数
        }
    }
    int ans=0;
    vector<vector<int>>dir={{0,1},{0,-1},{1,0},{-1,0}};
    while(!q.empty()){
        auto qq=q.front(); //取出当前元素
        q.pop();
        for(auto d:dir){
            int dx=qq.first+d[0];
            int dy=qq.second+d[1];
            if(dx<0||dx>=n||dy<0||dy>=m||grid[dx][dy]!=1) continue; //条件不满足不遍历
            grid[dx][dy]=grid[qq.first][qq.second]+1; //记录时间的重点
            cnt--;
            q.push(make_pair(dx,dy)); //相邻的橘子都被腐烂
            ans=grid[dx][dy]-2; //记录时间
            
        }
    }
    return cnt==0? ans:-1; //返回最长时间
}

---

做完这个题后再来练习一道相似的题目：P542 01矩阵

如果我们从每个1开始作为起点做广度优先搜索的化，会重复很多操作，因此我们考虑从0开始做多源广度优先搜索，跟上一题思想类似，注意的一点是：当我们搜索到这个1时，由于是从最近的0搜索到的，因此要将他的值置为0，防止再次搜索到他，代码如下：

vector<vector<int>> updateMatrix(vector<vector<int>>& mat) {
    int n=mat.size();
    int m=mat[0].size();
    queue<pair<int,int>>q;
    vector<vector<int>>ans(n,vector<int>(m));
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(mat[i][j]==0){
                ans[i][j]=0;
                q.push(make_pair(i,j));
            }
        }
    }
    vector<vector<int>>dir={{0,1},{0,-1},{1,0},{-1,0}};
    while(!q.empty()){
        auto qq=q.front();
        q.pop();
        for(auto d:dir){
            int dx=qq.first+d[0];
            int dy=qq.second+d[1];
            if(dx<0||dy<0||dx>=n||dy>=m||mat[dx][dy]==0) continue;
            ans[dx][dy]=ans[qq.first][qq.second]+1; //答案是前一个位置的值加1
            mat[dx][dy]=0; //修改1的值
            q.push(make_pair(dx,dy)); //将此1的位置入队列
        }
    }
    return ans;
}