【BZOJ2648】SJY摆棋子（KD-Tree）

题面

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

题解

$KD-Tree$ 模板题？
对于分割出来的矩阵，维护矩阵中的点的两维坐标的最大和最小值
这样就可以计算当前点到达这个矩阵的最短距离
这样就可以剪枝啦

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 555555
#define ls (t[x].ch[0])
#define rs (t[x].ch[1])
#define cmin(a,b) (a>b?a=b:a)
#define cmax(a,b) (a<b?a=b:a)
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
int D;
int n,rt,Q;
struct Node{int d[2];}a[MAX];
bool operator<(Node a,Node b){return a.d[D]<b.d[D];}
struct KD_Tree_Node
{
    int d[2],mn[2],mx[2];
    int ch[2];
}t[MAX<<1];
int ans;
void update(int x,int y)
{
    cmin(t[x].mn[0],t[y].mn[0]);cmin(t[x].mn[1],t[y].mn[1]);
    cmax(t[x].mx[0],t[y].mx[0]);cmax(t[x].mx[1],t[y].mx[1]);
}
int Build(int l,int r,int nD)
{
    int x=(l+r)>>1;D=nD;
    nth_element(&a[l],&a[x],&a[r+1]);
    t[x].d[0]=t[x].mn[0]=t[x].mx[0]=a[x].d[0];
    t[x].d[1]=t[x].mn[1]=t[x].mx[1]=a[x].d[1];
    if(l<x)ls=Build(l,x-1,nD^1),update(x,ls);
    if(r>x)rs=Build(x+1,r,nD^1),update(x,rs);
    return x;
}
int Dist(int nw,int X,int Y)
{
    return max(t[nw].mn[0]-X,0)+max(X-t[nw].mx[0],0)+max(t[nw].mn[1]-Y,0)+max(Y-t[nw].mx[1],0);
}
void Query(int x,int xx,int yy)
{
    int tmp=abs(t[x].d[0]-xx)+abs(t[x].d[1]-yy),d[2];
    d[0]=ls?Dist(ls,xx,yy):2e9;
    d[1]=rs?Dist(rs,xx,yy):2e9;
    cmin(ans,tmp);tmp=d[0]>=d[1];
    if(d[tmp]<ans)Query(t[x].ch[tmp],xx,yy);
    if(d[tmp^1]<ans)Query(t[x].ch[tmp^1],xx,yy);
}
int main()
{
    n=read();Q=read();
    for(int i=1;i<=n;++i)a[i].d[0]=read(),a[i].d[1]=read();
    rt=Build(1,n,0);
    while(Q--)
    {
        int opt=read(),x=read(),y=read();
        if(opt==1)
        {
            ++n;
            t[n].d[0]=t[n].mn[0]=t[n].mx[0]=x;
            t[n].d[1]=t[n].mn[1]=t[n].mx[1]=y;
            for(int nw=rt,d=0;nw;d^=1)
            {
                update(nw,n);
                int k=t[n].d[d]>=t[nw].d[d];
                if(!t[nw].ch[k]){t[nw].ch[k]=n;break;}
                nw=t[nw].ch[k];
            }
        }
        else
        {
            ans=2e9;Query(rt,x,y);
            printf("%d\n",ans);
        }
    }
    return 0;
}