Descripción del título
El mapa de la ciudad de Zhongshan es una matriz n * n, donde el número 1 indica un área comercial y el número 0 indica un área residencial. Para investigar la distancia entre el área residencial y el área comercial de la ciudad y hacer una evaluación de esto, el alcalde espera que pueda escribir un programa para llevar a cabo esta tarea. La distancia desde el área residencial i al área comercial se refiere a la distancia al área comercial más cercana j (| Xi-Xj | + | Yi-Yj |) (puede entender que sus rangos son diferentes), y usted Las estadísticas son la suma de las distancias desde todas las áreas residenciales a las áreas comerciales en el área de la matriz de (2 * r + 1) * (2 * r + 1) matriz centrada en ella para cada área k en la ciudad. El resultado también se emite en forma de matriz n * n.
De entrada
La primera línea es t, lo que indica que hay t conjuntos de datos a continuación, y cada conjunto de datos está separado por una línea en blanco. Lo siguiente: la
primera línea es n, r (1 <= r <n <= 150) y la
segunda línea es un n * La matriz de n.
Salida
t grupos de n * n matrices. Cada grupo está separado por una línea en blanco
Entrada de muestra
1
4 1
1 0 0 0
1 1 0 0
0 1 1 0
0 1 0 0
Salida de muestra
1 4 9 8
2 5 10 9
2 4 7 7
2 3 4 4
Análisis
Malditos datos del grupo, el juego rompió 0 y pensó que podía reírse demasiado violentamente. . .
Olvídate de las prácticas ultra violentas de Xian Kang Kang (70 puntos), ciclo de cinco veces 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;
}
Entonces ven a la solución positiva.
De hecho, puedes encontrar que bfs es una buena opción, puedes encontrar el 1 más cercano a la vez y marcar la distancia. Solo O (n 2 ).
Como se requiere la suma, el prefijo de intervalo y ans [i] [j] se usan para representar la suma de la matriz de (1,1) a (i, j), y luego la respuesta es qwq después de la resta. Presta atención a los límites especiales.
Código en
#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;
}