BZOJ 2648 SJY摆棋子(KD-Tree)

题目链接：BZOJ 2648  SJY摆棋子

题意：

在一个棋盘上，有N个黑色棋子。他每次要么放到棋盘上一个黑色棋子，要么放上一个白色棋子，如果是白色棋子，他会找出距离这个白色棋子最近的黑色棋子。此处的距离是 曼哈顿距离 即(|x1-x2|+|y1-y2|) 。现在给出N<=500000个初始棋子。和M<=500000个操作。对于每个白色棋子，输出距离这个白色棋子最近的黑色棋子的距离。同一个格子可能有多个棋子。

题解：

一个带插入的kd-tree搞之。

#include<cstdio>
#include<algorithm>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;

namespace KD_Tree{
    const int N=2e6+7,DI=2,inf=1<<30;
    struct Node{
        int p[DI],f,l,r,mx[DI],mi[DI];
        int operator[](const int&idx)const{return p[idx];}
        void in(int idx,int *v){f=idx;F(i,0,DI-1)p[i]=mi[i]=mx[i]=v[i];}
        void up(Node&a){
            F(i,0,DI-1){
                mi[i]=min(mi[i],a.mi[i]);
                mx[i]=max(mx[i],a.mx[i]);
            }
        }
    }T[N];
    int idx[N],cmpd,root,qx,qy,ans;
    
    bool cmp(const Node&a,const Node&b){
        return a[cmpd]!=b[cmpd]?a[cmpd]<b[cmpd]:a[!cmpd]<b[!cmpd];
    }
    void up(int x){
        if(T[x].l)T[x].up(T[T[x].l]);
        if(T[x].r)T[x].up(T[T[x].r]);
    }
    int build(int l,int r,int d=0,int f=0)
    {
        int mid=l+r>>1;
        cmpd=d%DI,nth_element(T+l,T+mid,T+r+1,cmp);
        idx[T[mid].f]=mid,T[mid].f=f;
        F(i,0,DI-1)T[mid].mi[i]=T[mid].mx[i]=T[mid][i];
        T[mid].l=l!=mid?build(l,mid-1,d+1,mid):0;
        T[mid].r=r!=mid?build(mid+1,r,d+1,mid):0;
        return up(mid),mid;
    }
    int cal(int x){return abs(T[x][0]-qx)+abs(T[x][1]-qy);}
    int dist(int k){
        if(!k)return inf;
        return max(T[k].mi[0]-qx,0)+max(qx-T[k].mx[0],0)+
        max(T[k].mi[1]-qy,0)+max(qy-T[k].mx[1],0);
    }
    int Query(int rt=root)
    {
        ans=min(ans,cal(rt));
        int dl=dist(T[rt].l),dr=dist(T[rt].r);
        int ls=T[rt].l,rs=T[rt].r;
        if(dl>dr)swap(ls,rs),swap(dl,dr);
        if(dl<ans)Query(ls);if(dr<ans)Query(rs);
    }
    int query(int x,int y){qx=x,qy=y,ans=inf,Query();return ans;}
    void insert(int k)
    {
        int o=root,d=1,ok=1,*ls,*rs;
        while(ok)
        {
            d^=1,T[o].up(T[k]),ls=&T[o].l,rs=&T[o].r;
            if(T[k][d]<=T[o][d])*ls?o=*ls:(*ls=k,ok=0);
            else *rs?o=*rs:(*rs=k,ok=0);
        }
    }
};

int t,n,m,x[2],op;
using namespace KD_Tree;

int main(){
    scanf("%d%d",&n,&m);
    F(i,1,n)scanf("%d%d",x,x+1),T[i].in(i,x);
    root=build(1,n);
    F(i,1,m)
    {
        scanf("%d%d%d",&op,x,x+1);
        if(op==1)T[i+n].in(i+n,x),insert(i+n);
        else printf("%d
",query(x[0],x[1]));
    }
    return 0;
}