显示图像(多源dfs)

问题 D: 显示图像

时间限制: 1 Sec 内存限制: 128 MB
提交状态

题目描述

古老的显示屏是由N×M个象素（Pixel）点组成的。一个象素点的位置是根据所在行数和列数决定的。例如P(2,1)表示第2行第1列的象素点。那时候，屏幕只能显示黑与白两种颜色，人们用二进制0和1来表示。0表示黑色，1表示白色。当计算机发出一个指令：P(x,y)=1，则屏幕上的第x行第y列的阴极射线管就开始工作，使该象素点显示白色，若P(x,y)=0，则对应位置的阴极射线管不工作，象素点保持黑色。在某一单位时刻，计算机以N×M二维01矩阵的方式发出显示整个屏幕图像的命令。
例如，屏幕是由3×4象素点组成，在某单位时刻，计算机发出如下命令：
0001
0011
0110
则屏幕图像为：
（假设放大后，一个格子表示一个象素点）

由于未知的原因，显示黑色的象素点总是受显示白色的象素点的影响——可能是阴极射线管工作的作用。并且，距离越近，影响越大。这里的距离定义如下：设有象素点P1(x1,y1)和象素点P2(x2,y2)，则它们之间的距离D(P1,P2)：D(P1,P2)=|x1-x2|+|y1-y2|
在某一时刻，计算机发出显示命令后，科学家们期望知道，每个象素点和其最近的显示白色的象素点之间的最短距离是多少——科学家们保证屏幕上至少有一个显示白色的象素点。
上面的例子中，象素P(1,1)与最近的白色象素点之间的距离为3，而象素P(3,2)本身显示白色，所以最短距离为0。

输入

第一行有两个数字，N和M (1<=N,M<=1000)，表示屏幕的规格。
以下N行，每行M个数字，0或1。为计算机发出的显示命令。

输出

输出有N行，每行M个数字，中间用1个空格分开。第i行第j列的数字表示距象素点P(i,j)最近的白色象素点的最短距离。

样例输入 Copy

样例输出 Copy

3 2 1 0
2 1 0 0
1 0 0 1

提示

对于100%的数据：N*M<=182^2。

#include<bits/stdc++.h> 
using namespace std;
typedef long long ll; 
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
#define pi 3.14159265358979323846
const int INF=0x3f3f3f3f;
const int mod=1e9+7;
const int maxn=1e6+100;
const int maxa=1e9+10;
struct node{
    int x,y;
};
int dx[4]={-1,1,0,0};
int dy[4]={0,0,-1,1};
int a[1010][1010];
int t[1010][1010];  
int n,m;
queue<node>q;
void inint(){
    cin>>n>>m;  int p;
    for(int i=0;i<=n;i++){
        for(int j=0;j<=m;j++){
            t[i][j]=1e9;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            scanf("%1d",&p);
            a[i][j]=p;
            if(p){
                t[i][j]=0;
                node o;
                o.x=i;
                o.y=j;
                q.push(o);
            }   
        }
    }
}
 
int main(){
    inint();
 
    while(!q.empty()){
        for(int i=0;i<=3;i++){
            int xx=q.front().x+dx[i];
            int yy=q.front().y+dy[i];
            if(xx>=1&&xx<=n&&yy>=1&&yy<=m){
                if(!a[xx][yy]&&t[q.front().x][q.front().y]+1<t[xx][yy]){
                    t[xx][yy]=t[q.front().x][q.front().y]+1;
                    node o;
                    o.x=xx;
                    o.y=yy;
                    q.push(o);
                }
            }
        }
        q.pop();
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            printf("%d ",t[i][j]);
        }
        printf("
");
    }                 
}