矩阵距离 —— 多源BFS模板

引入

在图论中，我们想求某个节点到很多节点的最短距离，

即求某个节点距离它最近的起点的距离是多少；

注意：不是多源最短路(floyd)，Floyd是求任意两个节点的；

我们可以将这个问题转化为单源最短路；

---

只需要构建一个虚拟源点；

将这个虚拟源点向每个起点连一条边权为0的边；

然后求某个点到虚拟源点的最短距离即可；

因为虚拟源点到我们这个点求最短路的时候会考虑每个分支；

那么虚拟源点求单源最短路和我们求距离最近的起点就是一回事了；

---

多源BFS

思想

思想和上面提到的建立虚拟源点类似；

但是我们并不需要去搞一个虚拟源点；

只需要认为它存在即可；

例题

矩阵距离

思路

根据上面提到的思想，我们先将所有为1的点赋值为0，并且塞入队列中，不断BFS即可；

Code

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <utility>
#include <queue>

using namespace std;

typedef long long ll;

const int N = 1e3 + 10;

int n,m;

char G[N][N];

typedef pair<int,int> pii;

queue<pii> que;

int dist[N][N];
int dx[] = {1,-1,0,0};
int dy[] = {0,0,1,-1};
void bfs(){
    for(int i=1;i<=n;++i){
        for(int j=1;j<=m;++j){
            if(G[i][j] == '1'){
                dist[i][j] = 0;
                que.push({i,j});
            }
        }
    }
    while(!que.empty()){
        auto u = que.front();
        que.pop();
        for(int i=0,xx,yy;i<4;++i){
            xx = dx[i] + u.first;
            yy = dy[i] + u.second;
            if(xx < 1 || xx > n || yy < 1 || yy > m) continue;
            if(dist[xx][yy] != -1) continue;
            dist[xx][yy] = dist[u.first][u.second] + 1;
            que.push({xx,yy});
        }
    }
}
void solve(){
    memset(dist,-1,sizeof dist);
    cin >> n >> m;
    for(int i=1;i<=n;++i) cin >> (G[i]+1);
    bfs();
    for(int i=1;i<=n;++i){
        for(int j=1;j<=m;++j){
            cout << dist[i][j] << ' ';
        }
        cout << '\n';
    }
}

int main(){
    std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
    solve();
    return 0;
}