纪中暑假培训：Date：9【GDKOI2004】城市统计 (Standard IO)

2137. 【GDKOI2004】城市统计 (Standard IO)

Time Limits: 1000 ms Memory Limits: 128000 KB Detailed Limits

Description

　　中山市的地图是一个n*n的矩阵，其中标号为1的表示商业区，标号为0的表示居民区。为了考察市内居民区与商业区的距离，并对此作出评估，市长希望你能够编写一个程序完成这一任务。
　　居民区i到商业区的距离指的是到距离它最近的商业区j的距离(|Xi-Xj|+|Yi-Yj|)，而你将统计的是对于城市中的每一个区域k，以它为中心，所有满足max(|Xk-Xm|,|Yk-Ym|)<=r的区域m到商业区距离之和。结果同样以n*n的矩阵形式输出。

Input

     第一行为t，表示以下有t组数据，每组数据之间以空行隔开，以下：

　　　第一行为n,r(1<=r

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>

using namespace std;
int map[155][155];
int dis[155][155];
int sum[155][155];
int ans;
int line[25001][3];
int fx[5],fy[5];
int n,m,k,head,tail,lx,ly,rx,ry;

bool check(int a,int b)
{
    if (a>=1&&a<=n&&b<=n&&b>=1&&!(dis[a][b])&&map[a][b]==0) return true;
    return false;
}

int main()
{
    scanf("%d",&k);
    fx[1]=1; fy[1]=0;
    fx[2]=0; fy[2]=1;
    fx[3]=-1; fy[3]=0;
    fx[4]=0; fy[4]=-1;
    for (int ii=1;ii<=k;ii++)
    {
        memset(map,0,sizeof(map));
        memset(dis,0,sizeof(dis));
        ans=0;
        scanf("%d%d",&n,&m);
        head=0;
        tail=1;
        for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            scanf("%d",&map[i][j]);
            if (map[i][j]==1)
            {
                tail++;
                line[tail][1]=i;
                line[tail][2]=j;
                dis[i][j]=0;
            }
        }
        while (head<=tail)
        {
            head++;
            for (int i=1;i<=4;i++)
            {
                int x,y;
                x=line[head][1]+fx[i];
                y=line[head][2]+fy[i];
                if (check(x,y))
                {
                    tail++;
                    line[tail][1]=x;
                    line[tail][2]=y;
                    dis[x][y]=dis[line[head][1]][line[head][2]]+1;  
                }
            }
        }
        for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        sum[i][j]=sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+dis[i][j];
        for (int i=1;i<=n;i++)
        {
        for (int j=1;j<=n;j++)
         {
             lx=max(i-m-1,0); ly=max(j-m-1,0);
             rx=min(i+m,n); ry=min(j+m,n);
             ans=sum[rx][ry]-sum[rx][ly]-sum[lx][ry]+sum[lx][ly];
             printf("%d ",ans);
         }
        printf("\n");
        }
        printf("\n");
    }
}