bzoj 2716 [Violet 3]天使玩偶——KDtree

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=2716

第三道KDtree!仍旧是模板。还有CDQ分治做法,见下面。

数组迷之开大?(开6e5+5还不行)

原来query( )只用传一个根。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=3e6+5,INF=0x3f3f3f3f;
int n,m,ans,rt,tot;
bool fx;
struct Dt{
    int x[2],y[2],p[2];
}a[N>>1];
bool cmp(Dt u,Dt v){return u.p[fx]<v.p[fx];}
struct KD{
    int c[N][2]; Dt s[N],q;
    void add(int cr,Dt k)
    {    for(int i=0;i<=1;i++) s[cr].x[i]=s[cr].y[i]=s[cr].p[i]=k.p[i];}
    void pshp(int cr)
    {
        int ls=c[cr][0],rs=c[cr][1];
        for(int i=0;i<=1;i++)
        {
            if(ls) s[cr].x[i]=min(s[cr].x[i],s[ls].x[i]),
                   s[cr].y[i]=max(s[cr].y[i],s[ls].y[i]);
            if(rs) s[cr].x[i]=min(s[cr].x[i],s[rs].x[i]),
                   s[cr].y[i]=max(s[cr].y[i],s[rs].y[i]);
        }
    }
    void build(int &cr,int l,int r,bool now)
    {
        int mid=l+r>>1; fx=now; nth_element(a+l,a+mid,a+r+1,cmp);
        cr=++tot;  add(cr,a[mid]);
        if(l<mid) build(c[cr][0],l,mid-1,!now);
        if(mid<r) build(c[cr][1],mid+1,r,!now);
        pshp(cr);
    }
    void insert(int &cr,bool now)
    {
        if(!cr){cr=++tot; add(cr,q); return;}
        if(q.p[now]<s[cr].p[now]) insert(c[cr][0],!now);
        else insert(c[cr][1],!now);
        pshp(cr);
    }
    int dis(int cr,Dt k){return abs(s[cr].p[0]-k.p[0])+abs(s[cr].p[1]-k.p[1]);}
    int dist(int cr,Dt k)
    {
        int ret=0;
        for(int i=0;i<=1;i++)
            ret+=max(0,s[cr].x[i]-k.p[i])+max(0,k.p[i]-s[cr].y[i]);
        return ret;
    }
    void query(int cr)
    {
        ans=min(ans,dis(cr,q));
        int ls=c[cr][0],rs=c[cr][1];
        int dl=(ls?dist(ls,q):INF),dr=(rs?dist(rs,q):INF);
        if(dl<dr) {if(dl<ans) query(ls); if(dr<ans) query(rs);}
        else {if(dr<ans) query(rs); if(dl<ans) query(ls);}
    }
}kd;
int rdn()
{
    int ret=0;char ch=getchar();
    while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') ret=(ret<<3)+(ret<<1)+ch-'0', ch=getchar();
    return ret;
}
int main()
{
    n=rdn(); m=rdn();
    for(int i=1;i<=n;i++) a[i].p[0]=rdn(), a[i].p[1]=rdn();
    kd.build(rt,1,n,0);
    for(int i=1,op;i<=m;i++)
    {
        op=rdn();kd.q.p[0]=rdn();kd.q.p[1]=rdn();
        if(op==1) kd.insert(rt,0);
        else { ans=INF;  kd.query(rt);  printf("%d\n",ans);}
    }
    return 0;
}

 CDQ分治做法可看看这个人的博客:https://www.cnblogs.com/longshengblog/p/5511593.html

  大约就是算距离的时候用 (x+y)-(x'+y') ,(x'<x,y'<=y) 。但只是这样只能算左下角的;所以用maxx-x作为新的x,y还是原来的,就能算右下角;用maxy-y作为新的y,x还是原来的,就能算左上角;x和y都减了,就能算右上角。那个博客的实现很简洁,所以自己就懒得写了(?)。

posted on 2018-09-06 17:01  Narh  阅读(167)  评论(0编辑  收藏  举报

导航