[POJ 2329] Nearest number-2

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>

using namespace std;

const int MAXN=200+10,INF=1<<27;
struct number{int cnt,d,x,y;}dp[MAXN][MAXN]; 
int n,dat[MAXN][MAXN];

void check(int a,int b,int l,int r)
{
    if(dp[a][b].d+1<dp[l][r].d) 
        dp[l][r]=dp[a][b],dp[l][r].d++;
    else if(dp[a][b].d+1==dp[l][r].d)  //注意这里的判断细节 
    {
        if(dp[l][r].cnt==1 && dp[a][b].cnt==1 && dp[l][r].x==dp[a][b].x && dp[l][r].y==dp[a][b].y) return;
        dp[l][r].cnt+=dp[a][b].cnt; //cnt都为1时才返回
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) for(int j=1;j<=n;j++)
        scanf("%d",&dat[i][j]),dp[i][j].d=INF;
    for(int i=1;i<=n;i++) for(int j=1;j<=n;j++)
    {
        if(dat[i][j]) dp[i][j].d=0,dp[i][j].x=i,dp[i][j].y=j,dp[i][j].cnt=1;
        check(i,j,i+1,j);check(i,j,i,j+1);            
    }
    for(int i=n;i>=1;i--) for(int j=n;j>=1;j--)
    {
        if(dat[i][j]) dp[i][j].d=0,dp[i][j].x=i,dp[i][j].y=j,dp[i][j].cnt=1;
        check(i,j,i-1,j);check(i,j,i,j-1);
    }
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            if(dat[i][j]){printf("%d ",dat[i][j]);continue;} 
            if(dp[i][j].cnt==1) printf("%d ",dat[dp[i][j].x][dp[i][j].y]);
            else printf("0 ");
        }
        puts("");
    }
    return 0;
}