代码如下
#include<bits/stdc++.h>
using namespace std;
int n,m;
const int N=8e2+5;
char maze[N][N];
int vis[N][N];
int dir[4][2]={0,1,0,-1,1,0,-1,0};
struct Node{ ///表示距离(x,y)点最近的距离为t
int x,y,t;
};
void bfs(){ ///多源BFS,将所有水域加入队列
queue<Node> q;
memset(vis,-1,sizeof vis);
for(int i=0;i<n;i++){ ///将所有水域加入队列
for(int j=0;j<m;j++){
if(maze[i][j]=='0') vis[i][j]=0,q.push({i,j,0});
}
}
while(q.size()){
Node now=q.front();q.pop();
for(int i=0;i<4;i++){
int xx=now.x+dir[i][0],yy=now.y+dir[i][1];
if(xx<0||xx>=n||yy<0||yy>=m||vis[xx][yy]!=-1||maze[xx][yy]==0) continue;
q.push({xx,yy,now.t+1}); ///下一个节点的步数要加一
vis[xx][yy]=now.t+1; ///记录下一个节点离水域的距离
}
}
}
int main(){
while(~scanf("%d%d",&n,&m)){
for(int i=0;i<n;i++) scanf("%s",&maze[i]); ///读入迷宫数据
bfs();
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
printf("%d ",vis[i][j]);
}
puts(""); ///输出换行
}
}
return 0;
}
总结:多源BFS问题和普通BFS问题的最大不同是普通BFS只将一个点加入队列,而多源BFS问题是将多个点加入队列(每个点的地位都是平等的),然后接着按套路处理,多源BFS问题类似于拓扑排序(拓扑排序是首先将度为零的点加入队列)。