DESCRIPTION
今天HHHH考完了期末考试,他在教学楼里闲逛,他看着教学楼里一间间的教室,于是开始思考:
如果从一个坐标为 (x1,y1,z1)(x1,y1,z1)的教室走到(x2,y2,z2)(x2,y2,z2)的距离为 |x1−x2|+|y1−y2|+|z1−z2||x1−x2|+|y1−y2|+|z1−z2|
那么有多少对教室之间的距离是不超过RR的呢?
INPUT
第一行是一个整数
T(1≤T≤10)T(1≤T≤10), 表示有
TT组数据接下来是
TT组数据,对于每组数据:第一行是两个整数
n,q(1≤n≤5×104,1≤q≤103)n,q(1≤n≤5×104,1≤q≤103), 表示有
nn间教室,
qq次询问.接下来是
nn行, 每行3个整数
xi,yi,zi(0≤xi,yi,zi≤10)xi,yi,zi(0≤xi,yi,zi≤10),表示这间教室的坐标.最后是
qq行,每行一个整数
R(0≤R≤109)R(0≤R≤109),意思见描述.
OUTPUT
对于每个询问
RR输出一行一个整数,表示有多少对教室满足题目所述的距离关系.
SAMPLE INPUT
13 30 0 01 1 11 1 1123
SAMPLE OUTPUT
113
HINT
对于样例,1号教室和2号教室之间的距离为3, 1号和3号之间的距离为3, 2号和3号之间的距离为0
SOLUTION
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define max_n 2000010
using namespace std;
typedef long long LL;
LL a[max_n],b[100],c[100];
struct node
{
int x;
int y;
int z;
}s[2000];
int main()
{
int t,p1,p2,p3;
LL n,q;
scanf("%d",&t);
while(t--)
{
int j=0;
memset(s,0,sizeof(s));
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
scanf("%lld %lld",&n,&q);
for(int i=0;i<n;i++)
{
scanf("%d %d %d",&p1,&p2,&p3);
int ans=p1*11+p2*200+p3*3000;
if(a[ans]==0)
{
s[j].x=p1,s[j].y=p2,s[j].z=p3;
j++;
}
a[ans]++;
}
for(int i=0;i<j;i++)
{
for(int p=i;p<j;p++)
{
int res1=s[i].x*11+s[i].y*200+s[i].z*3000;
int res2=s[p].x*11+s[p].y*200+s[p].z*3000;
if(i!=p)
{
int ans=abs(s[i].x-s[p].x)+abs(s[i].y-s[p].y)+abs(s[i].z-s[p].z);
b[ans]+=a[res1]*a[res2];
}
else
{
if(a[res1])//注意这个bug点;
{
b[0]+=a[res1]*(a[res1]-1)/2;
}
}
}
}
LL cnt=0,r;
for(int i=0;i<=35;i++)
{
cnt+=b[i];
c[i]=cnt;
}
while(q--)
{
scanf("%lld",&r);
if(r>=32) printf("%lld\n",c[32]);
else printf("%lld\n",c[r]);
}
}
return 0;
}