玲珑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;
}