玲珑杯 Round #18 A -- 计算几何你瞎暴力

因为坐标都在1-10之间，因此可暴力求解

#include <iostream>

#include <cstdio>

#include <cstring>

#include <queue>

#include <cmath>

#include <vector>

#include <algorithm>

using namespace std;

#define lowbit(x) (x&(-x))

#define max(x,y) (x>y?x:y)

#define min(x,y) (x<y?x:y)

#define mem(a) (memset(a,0,sizeof(a)))

typedef long long ll;

ll n,m,x,y,z,t;

ll a[][][];

ll b[];

ll sum[];

void init()

{

    for(ll i=;i<=;i++)

    {

        for(ll j=;j<=;j++)

        {

            for(ll k=;k<=;k++)

            {

                a[i][j][k]=;

            }

        }

    }

    for(ll i=;i<=;i++)

    {

        b[i]=;

        sum[i]=;

    }

}

void solve()

{

    for(ll i=;i<=;i++)

    {

        for(ll j=;j<=;j++)

        {

            for(ll k=;k<=;k++)

            {

                b[]+=((a[i][j][k]-)*a[i][j][k])/;

            }

        }

    }

    for(ll i=;i<=;i++)

    {

        for(ll j=;j<=;j++)

        {

            for(ll k=;k<=;k++)

            {

                for(ll x=;x<=;x++)

                {

                    for(ll y=;y<=;y++)

                    {

                        for(ll z=;z<=;z++)

                        {

                            if(i==x && j==y && z==k) continue;

                            ll t=abs(x-i)+abs(y-j)+abs(z-k);

                            b[t]+=(a[i][j][k]*a[x][y][z]);

                        }

                    }

                }

            }

        }

    }

    sum[]=b[];

    for(int i=;i<=;i++)

        sum[i]=sum[i-]+b[i]/;

}

int main()

{

    scanf("%lld",&t);

    while(t--)

    {

        init();

        scanf("%lld%lld",&n,&m);

        for(int i=;i<=n;i++)

        {

            scanf("%lld%lld%lld",&x,&y,&z);

            a[x][y][z]++;

        }

        solve();

        while(m--)

        {

            scanf("%lld",&x);

            if(x>) x=;

            printf("%lld\n",sum[x]);

        }

    }

    return ;

}