AcWing 1076. 迷宫问题 多源BFS

#include<bits/stdc++.h>
using namespace std;
const int N=1010;
#define x first
#define y second
typedef pair<int,int> pii;
int n,m,dis[N][N];;
char g[N][N];
void bfs(){
    queue<pii>q;
    int nx[4]={-1,1,0,0},ny[4]={0,0,1,-1};
    memset(dis,-1,sizeof dis);
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            if(g[i][j]=='1'){
                q.push({i,j});
                dis[i][j]=0;
            }
        }
    }
    while(!q.empty()){
        auto now=q.front();
        q.pop();
        for(int i=0;i<4;i++){
            int a=now.x+nx[i],b=now.y+ny[i];
            if(a<0||b<0||a>n-1||b>m-1)continue;
            if(dis[a][b]!=-1||g[a][b]=='1')continue;
            dis[a][b]=dis[now.x][now.y]+1;
            q.push({a,b});
        }
    }
}
int main(){
    cin>>n>>m;
    for(int i=0;i<n;i++){
        cin>>g[i];
    }
    bfs();
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cout<<dis[i][j]<<' ';
        }
        puts("");
    }
    return 0;
    
}