2020寒假【gmoj1598】【GDKOI2004】【城市统计city】【BFS+区间前缀和】

题目描述

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

输入

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

输出

t组n*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

分析

该死的t组数据，搞的比赛爆0还以为自己暴力能过在偷笑。。。
算了先康康超暴力做法吧（70pts），五重循环aa。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
using namespace std;
int t,n,r,a[160][160],f[160][160],ans,sum,dis;
int main()
{
	freopen("city.in","r",stdin);
	freopen("city.out","w",stdout);
	memset(f,127,sizeof(f));
    cin>>t;
    for(register int k=1;k<=t;k++)
    {
	    cin>>n>>r;
    	memset(a,0,sizeof(a));
    	memset(f,127,sizeof(f));
    	for(register int i=1;i<=n;i++)
	    {
	    	for(register int j=1;j<=n;j++)
	    	{
	    		scanf("%d",&a[i][j]);
	    		if(a[i][j]==1)
	    		{
	    			for(register int x=1;x<=n;x++)
	    			{
	    				for(register int y=1;y<=n;y++)
	    				{
	    					dis=abs(x-i)+abs(y-j);
	    					f[x][y]=min(f[x][y],dis);
						}
					}
				}
			}
		}
		for(register int i=1;i<=n;i++)
		{
			for(register int j=1;j<=n;j++)
			{
				sum=0;
				for(register int x=i-r;x<=i+r;x++)
				{
					for(register int y=j-r;y<=j+r;y++)
					{
						if(x>0&&y>0&&x<=n&&y<=n) sum+=f[x][y];
					}
				}
				a[i][j]=sum;
			}
		}
		for(register int i=1;i<=n;i++)
		{
			for(register int j=1;j<=n;j++)
			{
				printf("%d ",a[i][j]);
			}
			printf("\n");
		}
		printf("\n");
	}
	fclose(stdin);
	fclose(stdout);
    return 0;
}

然后再来说正解。
其实可以发现bfs是个很好的选择，一下子就可以找到最近的1，并标出距离。就O（n²）了。
因为要求和，所以就用区间前缀和ans[i][j]表示(1,1)到（i,j)这个矩阵的总和，然后减一下就出答案了qwq。注意特判边界。

上代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int t,n,r,f[160][160],ans[160][160],a[160][160]; 
const int dx[5]={0,1,0,-1,0};
const int dy[5]={0,0,-1,0,1};
queue<int> x,y;
void bfs()
{
	while(!x.empty())
	{
		for(int i=1;i<=4;i++)
		{
			int xx=x.front()+dx[i];
			int yy=y.front()+dy[i];
			if(xx>0&&xx<=n&&yy>0&&yy<=n&&a[xx][yy]==0)
			{
				a[xx][yy]=1;
				f[xx][yy]=f[x.front()][y.front()]+1;
				x.push(xx);
				y.push(yy);
			}
		}
		x.pop();
		y.pop();
	}
}
int main()
{
	freopen("city.in","r",stdin);
	freopen("city.out","w",stdout); 
    cin>>t;
    for(int k=1;k<=t;k++)
    {
    	memset(f,0,sizeof(f));
    	memset(ans,0,sizeof(ans));
    	cin>>n>>r;
    	for(int i=1;i<=n;i++)
    	{
    		for(int j=1;j<=n;j++) 
    		{
    			cin>>a[i][j];
    			if(a[i][j]==1)
    			{
    				x.push(i);
    				y.push(j);//压入队列 
				}
			} 
		}
		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]+f[i][j];
			}
			/*计算区间前缀和qwq*/
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				int lx=min(n,i+r),ly=min(n,j+r),rx=max(0,i-r-1),ry=max(0,j-r-1);
				cout<<ans[lx][ly]+ans[rx][ry]-ans[rx][ly]-ans[lx][ry]<<' ';
			}
			cout<<endl;
	    }
	    cout<<endl;
	}
	fclose(stdin);
	fclose(stdout);
    return 0;
}