（纪中）1598. 【GDKOI2004】城市统计（city）【BFS】

(File IO): input:city.in output:city.out
时间限制: 1000 ms 空间限制: 262144 KB 具体限制
Goto ProblemSet

题目描述
中山市的地图是一个$n\ast n$的矩阵，其中标号为$1$的表示商业区，标号为$0$的表示居民区。为了考察市内居民区与商业区的距离，并对此作出评估，市长希望你能够编写一个程序完成这一任务。 居民区$i$到商业区的距离指的是到距离它最近的商业区$j$的距离($|Xi-Xj|+|Yi-Yj|$)（你可以理解为他们的行列分别作差），而你将统计的是对于城市中的每一个区域$k$，以它为中心的$(2\ast r+1)\ast (2\ast r+1)$的矩阵区域内所有居民区到商业区的距离总和。结果同样以$n\ast n$的矩阵形式输出。

输入
第一行为$t$，表示以下有$t$组数据，每组数据之间以空行隔开，以下：
第一行为$n,r(1<=r<n<=150)$
第二行起为一个$n\ast n$的矩阵。

---

输出
$t$组$n\ast n$的矩阵。每组用空行隔开

---

样例输入
1
4 1
1 0 0 0
1 1 0 0
0 1 1 0
0 1 0 0

样例输出
1 4 9 8
2 5 10 9
2 4 7 7
2 3 4 4

---

数据范围限制

---

解题思路
根据题意，可分为两个阶段，首先对于给出的01图，计算出每个居民点到商业区的距离，接下来在进行求和。

计算距离
我们可以用一个队列，将所有的商业区放到队列里，同时将他们的距离标为$0$，然后从队列头开始，查看每一个结点的上下左右，如果有未标明距离的，且是居民点，就将其标为当前结点距离$+1$，然后把它将入队列，当队列里所有结点均已察看（也就是队列为空），所有点到商业区的距离就计算好了。

求和
然后是求和，如果按照一般的方法，枚举每一个点，然后以它为中心，将一个矩形内的所有点加起来求和，当数据很大的时候就会比较慢。
所以，我们设第一步计算距离完毕后的结果为矩阵$dis$，设$ans[i][j]$表示从矩阵$dis$左上角到$(i,j)$一个矩阵内所有点的和：先处理一下
$ans[i][j]=ans[i-1][j]+ans[i][j-1]-ans[i-1][j-1]+dis[i][j];$
那么若要求以$(i,j)$为中心，半径为$r$的正方形内所有点的和。
$s=ans[i+r][j+r]-ans[i-r-1][j+r]-ans[i+r][j-r-1]+ans[i-r-1][j-r-1]$
然后输出$s$就好

温馨提示：

1. 每输出一次答案要换行。
2. 求和时加上$r$或减去$r$是会超出矩阵$1..n$的范围。要注意处理一下以免越界。。。

---

代码

#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<algorithm>
#include<iomanip>
#include<cmath>
using namespace std;
const int dx[5]={0,1,-1,0,0};
const int dy[5]={0,0,0,1,-1};
int a[160][160],dis[160][160],l[25010][3],ans[160][160],T;
int ly,lx,ry,rx;
int h,t,n,s,r;
bool check(int x,int y){
     if(x>0&&y>0&&x<=n&&y<=n&&!(dis[x][y])&&a[x][y]==0)
	return 1;
     else
	return 0;
}
void bfs(){
     while(h<=t)
     {
	h++;
	for(int i=1;i<=4;i++)
	{
	     int x,y;
	     x=l[h][1]+dx[i];
	     y=l[h][2]+dy[i];
	     if(check(x,y))
	     {
			 t++;
		l[t][1]=x;
		l[t][2]=y;
		dis[x][y]=dis[l[h][1]][l[h][2]]+1;
	     }
	}
     }
}
int main(){
    freopen("city.in","r",stdin);
    freopen("city.out","w",stdout);
    scanf("%d",&T);
   for(int ii=1;ii<=T;ii++)
	{
		memset(a,0,sizeof(a));
		memset(dis,0,sizeof(dis));
		//memset(ans,0,sizeof(ans));
		//memset(l,0,sizeof(l));
	   s=0;
	   scanf("%d%d",&n,&r);
 	   h=0,t=1;
 	   for(int i=1;i<=n;i++)
  	   {
		for(int j=1;j<=n;j++)
		{
		    scanf("%d",&a[i][j]);
		    if(a[i][j]==1)
			{
				t++;
			l[t][1]=i;
			l[t][2]=j;
			dis[i][j]=0;
		    }
		}
   	  }
  	  bfs();
  	  for(int i=1;i<=n;i++)
		   for(int j=1;j<=n;j++)
	           ans[i][j]=ans[i-1][j]+ans[i][j-1]-ans[i-1][j-1]+dis[i][j];
  	  for(int i=1;i<=n;i++)
  	  {
		for(int j=1;j<=n;j++)
		{
		     lx=max(i-r-1,0); 
	 	    ly=max(j-r-1,0);      
		     rx=min(i+r,n); ry=min(j+r,n);            
	 	    s=ans[rx][ry]-ans[rx][ly]-ans[lx][ry]+ans[lx][ly];
    	           printf("%d ",s);
		}
		printf("\n");
	}
		printf("\n");
    }
}