[水题][2010Beijing wc]矩阵距离:广度优先搜索(bfs)

分析:

多源广搜。

代码:

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cctype>
#include <cstring>
#include <algorithm>
#include <queue> 
using namespace std;
int n,m,dx[5]={0,0,0,-1,1},dy[5]={0,-1,1,0,0},dis[1005][1005];
char s[1005][1005];
queue<int> q;
int main(){
    memset(dis,-1,sizeof dis);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        getchar();
        scanf("%s",s[i]+1);
        for(int j=1;j<=m;j++)
            if(s[i][j]=='1') q.push(i*1005+j),dis[i][j]=0;
    }
    while(!q.empty()){
        int x=q.front()/1005,y=q.front()%1005;q.pop();
        for(int i=1;i<=4;i++){
            int xx=x+dx[i],yy=y+dy[i];
            if(xx<1||xx>n||yy<1||yy>m||dis[xx][yy]!=-1) continue;
            dis[xx][yy]=dis[x][y]+1;
            q.push(xx*1005+yy);
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++)
            printf("%d ",dis[i][j]);
        printf("\n");
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ErkkiErkko/p/9595672.html