UVALive 7744 Finding Hotels（kd tree）

2016年青岛区域赛的k题，前两天做题碰到了，不会，抱着碰运气的心态分块来了一发，tle，又抱着侥幸的心态，把距离转化为切比雪夫距离，二分，然后爆搜了一发，还是tle。。。。搜了搜题解是kd tree，就学了学kd tree，这题也算是的裸的模板题。。。
这个模板在选取竖着分割还是横着分割，是根据所有点x坐标和y坐标的方差来选择的。
模板在：http://blog.csdn.net/zhjchengfeng5/article/details/7855241

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;

typedef long long LL;
const LL INF = 0x3fffffff;
const LL MAXN = 201000;

struct data
{
    LL pos[2];
    int id,price;
} T[MAXN], op, point;
int split[MAXN],now,n,demension;

LL ans;
double var[10];

bool cmp(data a,data b)
{
    return a.pos[split[now]]<b.pos[split[now]];
}

void build(int L,int R)
{
    if(L>R) return;

    int mid=(L+R)>>1;

    for(int pos=0; pos<demension; pos++)
    {
        double ave=var[pos]=0.0;
        for(int i=L; i<=R; i++)
            ave+=T[i].pos[pos];
        ave/=(R-L+1);
        for(int i=L; i<=R; i++)
            var[pos]+=(T[i].pos[pos]-ave)*(T[i].pos[pos]-ave);
        var[pos]/=(R-L+1);
    }

    split[now=mid]=0;
    for(int i=1; i<demension; i++)
        if(var[split[mid]]<var[i]) split[mid]=i;

    nth_element(T+L,T+mid,T+R+1,cmp);

    build(L,mid-1);
    build(mid+1,R);
}

void query(int L,int R)
{
    if(L>R) return;
    int mid=(L+R)>>1;

    LL dis=0;
    for(int i=0; i<demension; i++)
        dis+=(op.pos[i]-T[mid].pos[i])*(op.pos[i]-T[mid].pos[i]);

    if(dis < ans && T[mid].price <= op.price)
    {
        ans=dis;
        point=T[mid];
    }
    else if(dis == ans && T[mid].price <= op.price && T[mid].id < point.id)
        point = T[mid];

    LL radius=(op.pos[split[mid]]-T[mid].pos[split[mid]])*(op.pos[split[mid]]-T[mid].pos[split[mid]]);

    if(op.pos[split[mid]]<T[mid].pos[split[mid]])
    {
        query(L,mid-1);
        if(radius<=ans) query(mid+1,R);
    }
    else
    {
        query(mid+1,R);
        if(radius<=ans) query(L,mid-1);
    }
}

int main()
{
    int t,n,q;
    scanf("%d",&t);
    demension = 2;
    while(t--)
    {
        scanf("%d%d",&n,&q);
        for(int i=1; i<=n; i++)
        {
            scanf("%lld %lld %d",&T[i].pos[0],&T[i].pos[1],&T[i].price);
            T[i].id=i;
        }
        build(1,n);  //建树
        while(q--)
        {
            for(int i=0; i<demension; i++)
                scanf("%lld",&op.pos[i]);
            scanf("%d",&op.price);

            ans=(((LL)INF)*INF);
            point.id = INF;
            query(1,n);
            for(int i=0; i<demension; i++)
                printf("%lld ",point.pos[i]);
            printf("%d\n",point.price);
        }
    }
    return 0;
}