题目描述
中山市的地图是一个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(n2)了。
因为要求和,所以就用区间前缀和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;
}