hdu5809 Ants KD-tree+并查集

初次接触kd树，其实kd树就是将空间上的点分块，然后暴力查找，查找的时候用估价函数判断最值。主要还是靠模板。

题意：二维平面上有n(10W)个点，蚂蚁一开始只会出现在某个点上，然后向着直线距离最近的点走。有10W个询问，每次询问两个点，询问从这两个点出发的蚂蚁是否会相遇。

题解：因为每个点的蚂蚁的行动路线是固定的，所以只需要把所有路线相连的点放入一个并查集里就好了，为了找到最近点可以使用kd树查找，对于相同点直接在求距离的时候判断一下就好了。

#include <bits/stdc++.h>
using namespace std;
#define sint long long
#define maxn 600000
#define inf 0x3f3f3f3f3f3f3f3fLL
int D,n,m,rt;
struct node
{
    int d[2],mx[2],mn[2],l,r,id;
    friend bool operator < (node aa,node bb)
    {
        return aa.d[D]<bb.d[D];
    }
}p[maxn],ansP;
sint sqr(int x)
{
    return 1LL*x*x;
}
int px,py;
sint ans;
bool judge(node aa,node bb)
{
    if(aa.d[0]!=bb.d[0]) return aa.d[0]<bb.d[0];
    return aa.d[1]<bb.d[1];
}
struct tree
{
    node t[maxn],T;
    void update(int k)
    {
        int l=t[k].l,r=t[k].r;
        for(int i=0;i<2;i++)
        {
            t[k].mn[i]=t[k].mx[i]=t[k].d[i];
            if(l)t[k].mn[i]=min(t[k].mn[i],t[l].mn[i]);
            if(r)t[k].mn[i]=min(t[k].mn[i],t[r].mn[i]);
            if(l)t[k].mx[i]=max(t[k].mx[i],t[l].mx[i]);
            if(r)t[k].mx[i]=max(t[k].mx[i],t[r].mx[i]);
        }
    }
    sint dist(int x)
    {
        sint dis=0;
        if(px<t[x].mn[0])dis+=sqr(t[x].mn[0]-px);
        if(px>t[x].mx[0])dis+=sqr(px-t[x].mx[0]);
        if(py<t[x].mn[1])dis+=sqr(t[x].mn[1]-py);
        if(py>t[x].mx[1])dis+=sqr(py-t[x].mx[1]);
        return dis;
    }
    sint dis(node aa)
    {
        sint dis=sqr(aa.d[0]-px)+sqr(aa.d[1]-py);
        if(!dis) dis=inf;
        return dis;
    }
    int build(int l,int r,int now)
    {
        D=now;
        int mid=(l+r)>>1;
        nth_element(p+l,p+mid,p+r+1);
        t[mid]=p[mid];
        for(int i=0;i<2;i++) t[mid].mn[i]=t[mid].mx[i]=t[mid].d[i];
        if(l<mid)t[mid].l=build(l,mid-1,now^1);
        if(r>mid)t[mid].r=build(mid+1,r,now^1);
        update(mid);
        return mid;
    }
    void ask(int x)
    {
        sint dl,dr,d0=dis(t[x]);
        if(d0<ans||(d0==ans&&judge(t[x],ansP))) ans=d0,ansP=t[x];
        dl=t[x].l? dist(t[x].l):inf;
        dr=t[x].r? dist(t[x].r):inf;
        if(dl<dr)
        {
            if(dl<=ans)ask(t[x].l);
            if(dr<=ans)ask(t[x].r);
        }
        else
        {
            if(dr<=ans)ask(t[x].r);
            if(dl<=ans)ask(t[x].l);
        }
    }
    void cn()
    {
        memset(t,0,sizeof(t));
    }
}kd;
sint getP()
{
    ans=inf;
    kd.ask(rt);
    return ansP.id;
}
int fa[maxn];
int getfa(int x){return fa[x]==x?fa[x]:fa[x]=getfa(fa[x]);}
int main()
{
    int T,cas=0,n,q,x,y;
    scanf("%d",&T);
    while(T--)
    {
        cas++;
        printf("Case #%d:\n",cas);
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&x,&y);
            p[i].d[0]=x;
            p[i].d[1]=y;
            p[i].id=i;
        }
        rt=kd.build(1,n,0);

        for(int i=1;i<=n;i++)fa[i]=i;

        for(int i=1;i<=n;i++)
        {
            px=p[i].d[0];
            py=p[i].d[1];
            x=p[i].id,y=getP();
            //printf("x: %d\n",x);
            //printf("len %d\n",y);
            //printf("link %d %d\n",getfa(x),getfa(y));
            fa[getfa(x)]=getfa(y);
            //getfa(x);getfa(y);
            //fa[getfa(x)]=getfa(y);
        }
        while(q--)
        {
            scanf("%d%d",&x,&y);
            puts(getfa(x)==getfa(y)?"YES":"NO");
        }
        kd.cn();
    }
    return 0;
}
/*
2
2 1
0 0
-1 1
1 2
5 2
1 1
3 3
4 4
0 -3
0 -4
1 3
2 4
*/