HDU 5992 Finding Hotels KDtree

    题意是说有T组样例输入，然后输入N和M，N是旅馆的个数，M是游客的个数。假设旅馆中来多少人都能住开，接下来N行分别是旅馆的位置x,y和旅馆的价格c，然后M行分别是这些人现在的位置和他们所带的钱。要求输出离每个人最近且这个人能付起这个旅馆房费的旅馆（只要付得起就行，不用考虑价格高低）。如果有很多旅馆同时满足要求，就输出样例中先输入进去的那一个旅馆。

    这个题几乎是KD树的一个模板题，只是多加了价格的限制以及最后处理一下旅馆顺序进行输出。抛开这些就是一个求最近距离的KD树问题。也是之前从来没见过这个算法，看了一下，其实和线段树也差不多，只是在处理上这是一个多维的二叉搜索树，进行多维查询，具体的就不多说了，因为对于KD树我自己其实也没太深入看懂= =临近区域赛，从网上找了份题解先看懂了，记录一下为比赛准备模板~

    下面AC代码：

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<climits>using namespace std;long long n,m;long long cmp_d;long long root;struct node{    long long l,r;    long long d[3],Max[3],Min[3];    long long id;};node a[200005];bool cmp(node a,node b){    return a.d[cmp_d]<b.d[cmp_d];}int up(long long p,long long k){    int i;    for(i=0;i<3;i++)    {        a[p].Min[i]=min(a[p].Min[i],a[k].Min[i]);        a[p].Max[i]=max(a[p].Max[i],a[k].Max[i]);    }    return 0;}long long build(long long l,long long r,long long D){    int i;    cmp_d=D;    long long mid=(l+r)/2;    nth_element(a+1+l,a+1+mid,a+1+r,cmp);    for(i=0;i<3;i++)        a[mid].Min[i]=a[mid].Max[i]=a[mid].d[i];    if(l!=mid)        a[mid].l=build(l,mid-1,(D+1)%3);    else        a[mid].l=0;    if(r!=mid)        a[mid].r=build(mid+1,r,(D+1)%3);    else        a[mid].r=0;    if(a[mid].r)        up(mid,a[mid].r);    if(a[mid].l)        up(mid,a[mid].l);    return mid;}long long x,y,z;long long jl,ans;long long getdis(long long p){    long long res=0;    if(z<a[p].Min[2])        return LLONG_MAX;    if(x>a[p].Max[0])        res+=(x-a[p].Max[0])*(x-a[p].Max[0]);    if(x<a[p].Min[0])        res+=(a[p].Min[0]-x)*(a[p].Min[0]-x);    if(y>a[p].Max[1])        res+=(y-a[p].Max[1])*(y-a[p].Max[1]);    if(y<a[p].Min[1])        res+=(a[p].Min[1]-y)*(a[p].Min[1]-y);    return res;}int ask(long long p){    long long d0=0;    long long dl,dr;    if(a[p].d[2]>z)        d0+=LLONG_MAX;    else    {        d0+=(a[p].d[0]-x)*(a[p].d[0]-x)+(a[p].d[1]-y)*(a[p].d[1]-y);        if(d0<jl)        {            ans=p;            jl=d0;        }        else if(d0==jl)        {            if(a[p].id<a[ans].id)            {                ans=p;            }        }    }    if(a[p].l)        dl=getdis(a[p].l);    else        dl=LLONG_MAX;    if(a[p].r)        dr=getdis(a[p].r);    else        dr=LLONG_MAX;    if(dl<dr)    {        if(dl<=jl)            ask(a[p].l);        if(dr<=jl)            ask(a[p].r);    }    else    {        if(dr<=jl)            ask(a[p].r);        if(dl<=jl)            ask(a[p].l);    }    return 0;}int main(){    int T;    int i,j;    scanf("%d",&T);    while(T--)    {        scanf("%lld%lld",&n,&m);        for(i=1;i<=n;i++)        {            for(j=0;j<3;j++)            {                scanf("%lld",&a[i].d[j]);            }            a[i].l=a[i].r=0;            a[i].id=i;        }        root=build(1,n,0);        for(i=1;i<=m;i++)        {            scanf("%lld%lld%lld",&x,&y,&z);            jl=LLONG_MAX;            ans=-1;            ask(root);            cout<<a[ans].d[0]<<" "<<a[ans].d[1]<<" "<<a[ans].d[2]<<endl;        }    }    return 0;}