多源BFS

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define x first
#define y second
using namespace std;
const int N = 1010;
typedef pair<int,int> PII;
int n,m;
int dist[N][N];
char g[N][N];
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};

void bfs()
{
    memset(dist,-1,sizeof dist);
    queue<PII> q;
    for(int i = 0 ; i < n ; i++)
        for(int j = 0  ; j < m ; j++)
        {
            if(g[i][j]=='1')
            {
                dist[i][j] = 0 ;
                q.push({i,j});
            }
        }

    while(q.size())
    {
        PII t  = q.front();
        q.pop();
        for(int i = 0 ; i < 4 ; i++)
        {
            int a = t.x+dx[i];
            int b = t.y+dy[i];
            if(a < 0 || a >= n || b < 0 || b >= m || dist[a][b]!=-1) continue;
            dist[a][b] = dist[t.x][t.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++)
            printf("%d ",dist[i][j]);
        puts("");
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值