Problem Description
今天H考完了期末考试,他在教学楼里闲逛,他看着教学楼里一间间的教室,于是开始思考:
如果从一个坐标为 (x1,y1,z1)的教室走到(x2,y2,z2))的距离为 |x1−x2|+|y1−y2|+|z1−z2|
那么有多少对教室之间的距离是不超过R的呢?
Input
第一行是一个整数T(1≤T≤10), 表示有T组数据 接下来是T组数据,对于每组数据: 第一行是两个整数n,q(1≤n≤5×104,1≤q≤103)表示有n间教室, q次询问. 接下来是n行, 每行3个整数xi,yi,zi(0≤xi,yi,zi≤10)表示这间教室的坐标. 最后是q行,每行一个整数R(0≤R≤109),意思见描述.
Output
对于每个询问R输出一行一个整数,表示有多少对教室满足题目所述的距离关系.
Sample Input
1
3 3
0 0 0
1 1 1
1 1 1
1
2
3
Sample Output
1
1
3
Hint
对于样例,1号教室和2号教室之间的距离为3, 1号和3号之间的距离为3, 2号和3号之间的距离为0
题意:在一个三维空间中有N个点,q次查询,每次查询给一距离r,求出三维空间中有多少对点之间的哈密顿距离小于r。
思路:一开始的时候如果按最简单的想法,先两两配对求出每两个点之间的距离,之后输出,但是本题中点的数目n的数据较大,如果要全部处理的话需要10^9的复杂度,肯定会超时。这个时候我们观察后发现,每一个点的范围很小,0<=x,y,z<=10,如果我们通过坐标来遍历每一个点,那么就只需要10^3的复杂度,显然更合适。所以本题也是如此,通过以坐标为单位的枚举,就可以得到最后的结果:
注释:哈密顿距离定义是:对于二维平面上的两点A,B,其坐标分别为:,,那么它们的哈密顿距离D表示为:,然后对x维度的(x1-x2)用fabs或者abs函数取绝对值,再对y维度取绝对值
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
int n, q, t, tem;
int a, b, c, i, x, y, z;
long long int aa[35];
long long int d[15][15][15];
int dis(int a, int b, int c, int x, int y, int z)
{
return fabs(a-x)+fabs(b-y)+fabs(c-z);
}
int main()
{
scanf("%d",&t);
while(t--)
{
memset(aa, 0, sizeof(aa));
memset(d, 0, sizeof(d));
scanf("%d%d",&n,&q);
while(n--)
{
scanf("%d%d%d",&x,&y,&z);
d[x][y][z]++;
}
for(a = 0; a <= 10; a++)
{
for(b = 0; b <= 10; b++)
{
for(c = 0; c <= 10; c++)
{
if(d[a][b][c])
{
for(x = 0; x <= 10; x++)
{
for(y = 0; y <= 10; y++)
{
for(z = 0; z <= 10; z++)
{
if(d[x][y][z])
{
tem = dis(a, b, c, x, y, z);
if(tem == 0)
aa[tem] += (d[x][y][z])*(d[x][y][z]-1)/2;
else
aa[tem] += d[x][y][z]*d[a][b][c];
}
}
}
}
}
}
}
}//只枚举x,y,z在范围内,输入的每两个点之间的距离
for(i = 1; i <= 30; i++)
aa[i] /= 2;
for(i = 1; i <= 30; i++)
aa[i] += aa[i-1];
while(q--)
{
scanf("%d",&tem);
if(tem > 30)
tem = 30;
printf("%lld\n",aa[tem]);
}
}
return 0;
}
转载自博客园 作者87hbteo 略有改动....