矩阵距离(多源bfs)

题目:

题目链接:

题解:

#include <bits/stdc++.h>
using namespace std;
int n,m;
char pos[1005][1005];
int dis[1005][1005];
int dx[4]={0,0,1,-1};
int dy[4]={1,-1,0,0};
typedef pair<int,int> pll;
void bfs()
{
    memset(dis,-1,sizeof dis);
    queue<pll> q;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            if(pos[i][j]=='1')
            {
                dis[i][j]=0;
                q.push({i,j});
            }
        }
    }
    while(!q.empty())
    {
        pll k=q.front();
        q.pop();
        int x=k.first,y=k.second;
        for(int i=0;i<4;i++)
        {
            int nx=x+dx[i],ny=y+dy[i];
            if(nx<0||nx>=n||ny<0||ny>=m||dis[nx][ny]!=-1) continue;
            dis[nx][ny]=dis[x][y]+1;
            q.push({nx,ny});
        }
    }
}
int main()
{
    cin>>n>>m;
    for(int i=0;i<n;i++) cin>>pos[i];
    bfs();
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m;j++)
        {
            cout<<dis[i][j]<<" ";
        }
        cout<<endl;
    }
    return 0;
}