“玲珑杯”ACM比赛 Round #18 A -- 计算几何你瞎暴力（暴力）

1143 - 计算几何你瞎暴力

Time Limit：5s Memory Limit：256MByte

Submissions：1615Solved：306

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

13 30 0 01 1 11 1 1123

SAMPLE OUTPUT

113

HINT

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

n*n肯定要超时，分析下数据，发现坐标很小，可以直接通过计算每个点坐标个数，暴力枚举组合数

相同的时候考虑错公式了，应该是X*(X-1),写成了x-1

#include<string.h>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<queue>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int N=5005;
typedef long long ll;
struct node
{
    ll x,y,z;
}e[N];
ll Map[105][105][105];
ll dis[35];
ll ans[35];
ll Distance(node a,node b)
{
    return abs(a.x-b.x)+abs(a.y-b.y)+abs(a.z-b.z);
}
int main()
{
    ll t;
    scanf("%lld",&t);
    while(t--)
    {
        ll n,m,cnt=0,x,y,z,tt,sum=0;
        scanf("%lld%lld",&n,&m);
        mem(Map,0);
        mem(dis,0);
        mem(ans,0);
        for(ll i=1; i<=n; i++)
        {
            scanf("%lld%lld%lld",&x,&y,&z);
            if(!Map[x][y][z])
                e[cnt].x=x,e[cnt].y=y,e[cnt++].z=z,Map[x][y][z]++;
            else
                Map[x][y][z]++;
        }
        for(ll i=0; i<cnt; i++)
        {
            for(ll j=i; j<cnt; j++)
            {
                if(i!=j)
                {
                    ll temp=Distance(e[i],e[j]);
                    ll tx=Map[e[i].x][e[i].y][e[i].z];
                    ll ty=Map[e[j].x][e[j].y][e[j].z];
                    dis[temp]+=tx*ty;
                }
                else
                {
                    if(Map[e[i].x][e[i].y][e[i].z])
                     {
                         dis[0]+=Map[e[i].x][e[i].y][e[i].z]*(Map[e[i].x][e[i].y][e[i].z]-1)/2;
                     }
                }
            }
        }
        for(ll i=0; i<=30; i++)
        {
            sum+=dis[i];
            ans[i]=sum;
        }
        for(ll i=1; i<=m; i++)
        {
            scanf("%lld",&tt);
            if(tt>30)
                printf("%lld\n",ans[30]);
            else
                printf("%lld\n",ans[tt]);
        }

    }
}