bzoj2648/2716 kdtree

SJY摆棋子

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 5199  Solved: 1813
[Submit][Status][Discuss]

Description

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

 

Input

第一行两个数 N M

以后M行，每行3个数 t x y

如果t=1 那么放下一个黑色棋子

如果t=2 那么放下一个白色棋子

Output

对于每个T=2 输出一个最小距离

 

Sample Input

2 3
1 1
2 3
2 1 2
1 3 3
2 4 2

Sample Output

1
2

HINT

kdtree可以过

Source

鸣谢 孙嘉裕

 

题解：kdtree的应用，查询最近的距离，就是缩小范围，和最优性剪枝即可过了。

 #include<cstring>
 #include<cmath>
 #include<iostream>
 #include<algorithm>
 #include<cstdio>

 #define inf 1000000007
 #define N 500007
 #define ll long long
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if (ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=(x<<)+(x<<)+ch-'';ch=getchar();}
     return x*f;
 }

 int n,m,rt,F;
 struct Node
 {
     int d[],mn[],mx[],l,r;
     int& operator[](int x)
     {
         return d[x];
     }
     Node(int x=,int y=)//无代入的话x,y为0，代入即为代入值。
     {
         l=,r=;
         d[]=x,d[]=y;
     }
 }p[N];
 bool operator<(Node x,Node y)
 {
     return x[F]<y[F];
 }
 inline int dis(Node x,Node y)
 {
     return abs(x[]-y[])+abs(x[]-y[]);
 }
 struct kdtree
 {
     int ans;
     Node tr[N*],T;
     void update(int p)
     {
         Node l=tr[tr[p].l],r=tr[tr[p].r];
         for (int i=;i<;i++)
         {
             if (tr[p].l)tr[p].mn[i]=min(tr[p].mn[i],l.mn[i]),tr[p].mx[i]=max(tr[p].mx[i],l.mx[i]);
             if (tr[p].r)tr[p].mn[i]=min(tr[p].mn[i],r.mn[i]),tr[p].mx[i]=max(tr[p].mx[i],r.mx[i]);
         }
     }
     int build(int l,int r,int now)
     {
         F=now;int mid=(l+r)>>;
         nth_element(p+l,p+mid,p+r+);
         tr[mid]=p[mid];
         for (int i=;i<;i++)
             tr[mid].mn[i]=tr[mid].mx[i]=tr[mid][i];
         if (l<mid)tr[mid].l=build(l,mid-,now^);
         if (r>mid)tr[mid].r=build(mid+,r,now^);
         update(mid);
         return mid;
     }
     void ins(int p,int now)
     {
         if (T[now]>=tr[p][now])
         {
             if(tr[p].r)ins(tr[p].r,now^);
             else
             {
                 tr[p].r=++n,tr[n]=T;
                 for (int i=;i<;i++)
                     tr[n].mn[i]=tr[n].mx[i]=tr[n][i];
             }
         }
         else
         {
             if (tr[p].l)ins(tr[p].l,now^);
             else
             {
                 tr[p].l=++n,tr[n]=T;
                 for (int i=;i<;i++)
                     tr[n].mn[i]=tr[n].mx[i]=tr[n][i];
             }
         }
         update(p);
     }
     int get(int k,Node p)
     {
         int tmp=;
         for (int i=;i<;i++)
             tmp+=max(,tr[k].mn[i]-p[i]);
         for (int i=;i<=;i++)
              tmp+=max(,p[i]-tr[k].mx[i]);
         return tmp;
     }
     void query(int p,int now)
     {
         int d,dl=inf,dr=inf;
         d=dis(tr[p],T);
         ans=min(ans,d);
         if (tr[p].l)dl=get(tr[p].l,T);
         if (tr[p].r)dr=get(tr[p].r,T);
         if (dl<dr)
         {
             if (dl<ans)query(tr[p].l,now^);
             if (dr<ans)query(tr[p].r,now^);
         }
         else
         {
             if (dr<ans)query(tr[p].r,now^);
             if (dl<ans)query(tr[p].l,now^);
         }
     }
     int query(Node p)
     {
         ans=inf,T=p,query(rt,);
         return ans;
     }
     void ins(Node p)
     {
         T=p;
         ins(rt,);
     }
 }kd;
 int main()
 {
     freopen("fzy.in","r",stdin);
     freopen("fzy.out","w",stdout);

     n=read(),m=read();
     for (int i=;i<=n;i++)
         p[i][]=read(),p[i][]=read();
     rt=kd.build(,n,);//后者0，代表维的循环
     while(m--)
     {
         int flag=read(),x=read(),y=read();
         if (flag==) kd.ins(Node(x,y));
         else printf("%d\n",kd.query(Node(x,y)));
     }
 }