玲珑oj1143 计算几何你瞎暴力

1143 - 计算几何你瞎暴力

Time Limit：5s Memory Limit：256MByte

Submissions：1786Solved：354

DESCRIPTION

今天$HH$考完了期末考试,他在教学楼里闲逛,他看着教学楼里一间间的教室,于是开始思考:

如果从一个坐标为 $(x_1,y_1,z_1)$的教室走到$(x_2,y_2,z_2)$的距离为$|x_1-x_2|+|y_1-y_2|+|z_1-z_2|$

那么有多少对教室之间的距离是不超过$R$的呢?

INPUT

第一行是一个整数$T(1\le T\le 10)$, 表示有$T$组数据接下来是$T$组数据,对于每组数据:第一行是两个整数$n,q(1\le n\le 5\times {10}^4,1\le q\le {10}^3)$, 表示有$n$间教室,$q$次询问.接下来是$n$行, 每行3个整数$x_i,y_i,z_i(0\le x_i,y_i,z_i\le 10)$,表示这间教室的坐标.最后是$q$行,每行一个整数$R(0\le R\le {10}^9)$,意思见描述.

OUTPUT

对于每个询问$R$输出一行一个整数,表示有多少对教室满足题目所述的距离关系.

SAMPLE INPUT

1

3 3

0 0 0

1 1 1

1 1 1

1

2

3

SAMPLE OUTPUT

113

HINT

对于样例,1号教室和2号教室之间的距离为3, 1号和3号之间的距离为3, 2号和3号之间的距离为0

题解思路:考虑到坐标的范围非常小，我们可以统计每一个点有多少个教室，然后枚举坐标预处理答案，然后做一个前缀和即可，要小心R的范围；

代码:

#include<cstdio>#include<cstring>#include<algorithm>using namespace std;typedef long long ll;int pos[15][15][15];struct node{    int x,y,z;} p[11000];ll a[50],sum[50];int dis(int a,int b){    int res=a-b;    return res>0?(a-b):(b-a);}ll C(int a,int b){    int k=b;    ll sum=1;    while(k!=0)        sum*=a,a--,k--;    while(b!=0)        sum/=b,b--;    return sum;}int main(){    int T;    scanf("%d",&T);    while(T--)    {        memset(pos,0,sizeof(pos));        memset(sum,0,sizeof(sum));        memset(p,0,sizeof(p));        memset(a,0,sizeof(a));        int n,q;        int tot=0;        scanf("%d%d",&n,&q);        for(int i=0; i<n; i++)        {            int x,y,z;            scanf("%d%d%d",&x,&y,&z);            if(pos[x][y][z]==0)            {                p[tot].x=x,p[tot].y=y,p[tot].z=z;                tot++;            }            pos[x][y][z]++;        }        //枚举坐标，预处理；        for(int i=0;i<tot;i++)        {            for(int j=i+1;j<tot;j++)            {                int R=dis(p[i].x,p[j].x)+dis(p[i].y,p[j].y)+dis(p[i].z,p[j].z);                ll num=pos[p[i].x][p[i].y][p[i].z]*pos[p[j].x][p[j].y][p[j].z];                a[R]+=num;            }        }        //特别处理当距离为0的数量；        for(int i=0;i<tot;i++)        {            int n=pos[p[i].x][p[i].y][p[i].z];            if(n>=2)                a[0]+=C(n,2);        }//        for(int i=0;i<5;i++)//            printf("%d %d\n",i,a[i]);        //预处理前缀和；        sum[0]=a[0];        for(int i=1;i<50;i++)            sum[i]=sum[i-1]+a[i];        for(int i=0;i<q;i++)        {            int ask;            scanf("%d",&ask);            if(ask<=30)                printf("%d\n",sum[ask]);            else                printf("%d\n",sum[30]);        }    }    return 0;}