CodeChef COT5 线段树

题意

给出一些点，每个点有一个权值和键值，将这些点建成一个treap，支持插入一个点，删除一个点，询问两点之间的距离。

首先考虑怎么建树，将所有点按键值从小到大排序，每次从当前区间找出权值最大的点，作为当前子树的根，该点左边的点建成左子树，右边的点建成右子树。

把所有点离线下来，按键值排序，树上两点的lca就是在序列上两点之间的权值最大的点。找到lca后求两点距离就只需求一个点到根的距离就行了。

从一个点分别向左右每次找第一个比当前点键值大的点就是该点的祖先，然后就可以求距离了。

这玩意咋求？ @bzoj2957 线段树维护，复杂度nlog^2

第一次写博客，感觉自己在扯淡。。。算了，代码

#include <bits/stdc++.h>
using namespace std;
#define N 210000
#define ll unsigned int
#define ls l,mid,now<<1
#define rs mid+1,r,now<<1|1
map<ll,ll>ma;
struct node
{
    ll k,w;
    node(){}
    node(ll k,ll w):k(k),w(w){}
    friend bool operator < (const node &r1,const node &r2)
    {
        if(r1.k==r2.k)return r1.w<r2.w;
        return r1.k<r2.k;
    }
    friend bool operator == (const node &r1,const node &r2)
    {return r1.k==r2.k&&r1.w==r2.w;};
}a[N];
struct poi
{
    int num,n1,l,r,now;
    ll h;
}pl[N<<2],pr[N<<2];
ll mx[N<<2];
int n,m;
int type[N],r1[N],r2[N],p1[N],p2[N],pos[N<<2];
void build(int l,int r,int now)
{
    pl[now].l=pr[now].l=l;
    pl[now].r=pr[now].r=r;
    pl[now].now=pr[now].now=now;
    if(l==r)return;
    int mid=(l+r)>>1;
    build(ls);build(rs);
}
int getl(int l,int r,int now,ll x)
{
    if(l==r)
        return pl[now].h>x;
    int mid=(l+r)>>1;
    if(pl[now<<1].h>x)return getl(ls,x)+pl[now].n1;
    return getl(rs,x);
}
int getr(int l,int r,int now,ll x)
{
    if(l==r)
        return pr[now].h>x;
    int mid=(l+r)>>1;
    if(pr[now<<1|1].h>x)return getr(rs,x)+pr[now].n1;
    return getr(ls,x);
}
poi updl(poi l,poi r,int now)
{
    poi ret;
    ret.l=l.l;ret.r=r.r;
    ret.now=now;
    ret.n1=getl(r.l,r.r,r.now,l.h);
    ret.h=max(r.h,l.h);
    ret.num=l.num+ret.n1;
    return ret;
}
poi updr(poi l,poi r,int now)
{
    poi ret;
    ret.l=l.l;ret.r=r.r;
    ret.now=now;
    ret.n1=getr(l.l,l.r,l.now,r.h);
    ret.h=max(r.h,l.h);
    ret.num=r.num+ret.n1;
    return ret;
}
void pushup(int now)
{
    pl[now]=updl(pl[now<<1],pl[now<<1|1],now);
    pr[now]=updr(pr[now<<1],pr[now<<1|1],now);
    pos[now]=mx[now<<1]>mx[now<<1|1] ? pos[now<<1]:pos[now<<1|1];
    mx[now]=max(mx[now<<1],mx[now<<1|1]);
}
void update(int l,int r,int now,int x,ll y)
{
    if(l==r)
    {
        if(y)
        {
            pl[now].h=pr[now].h=y;
            pl[now].num=pr[now].num=1;
            mx[now]=y;pos[now]=l;
        }
        else
        {
            pl[now].h=pr[now].h=0;
            pl[now].num=pr[now].num=0;
            mx[now]=pos[now]=0;
        }
        return;
    }
    int mid=(l+r)>>1;
    if(mid>=x)update(ls,x,y);
    else update(rs,x,y);
    pushup(now);
}
int querym(int l,int r,int now,int lq,int rq)
{
    if(lq<=l&&r<=rq)
        return now;
    int mid=(l+r)>>1;
    if(mid<lq)return querym(rs,lq,rq);
    if(mid>=rq)return querym(ls,lq,rq);
    int t1=querym(ls,lq,rq);
    int t2=querym(rs,lq,rq);
    return mx[t1]>mx[t2] ? t1:t2;
}
poi queryl(int l,int r,int now,int lq,int rq)
{
    if(lq<=l&&r<=rq)
        return pl[now];
    int mid=(l+r)>>1;
    if(mid<lq)return queryl(rs,lq,rq);
    if(mid>=rq)return queryl(ls,lq,rq);
    poi t1=queryl(ls,lq,rq);
    poi t2=queryl(rs,lq,rq);
    return updl(t1,t2,0);
}
poi queryr(int l,int r,int now,int lq,int rq)
{
    if(lq<=l&&r<=rq)
        return pr[now];
    int mid=(l+r)>>1;
    if(mid<lq)return queryr(rs,lq,rq);
    if(mid>=rq)return queryr(ls,lq,rq);
    poi t1=queryr(ls,lq,rq);
    poi t2=queryr(rs,lq,rq);
    return updr(t1,t2,0);
}
int get(int x)
{
    poi t1=queryl(1,n,1,x,n);
    poi t2=queryr(1,n,1,1,x);
    return t1.num+t2.num;
}
int main()
{   
    //freopen("tt.in","r",stdin);
    scanf("%d",&m);
    for(int i=1;i<=m;i++) 
    {
        scanf("%d",&type[i]);
        if(type[i]==0)
        {
            scanf("%d%d",&r1[i],&r2[i]);
            a[++n]=node(r1[i],r2[i]);
        }
        if(type[i]==1)
            scanf("%d",&r1[i]);
        if(type[i]==2)
            scanf("%d%d",&r1[i],&r2[i]);
    }
    sort(a+1,a+1+n);
    n=unique(a+1,a+1+n)-a-1;
    build(1,n,1);
    for(int i=1;i<=m;i++)
    {
        if(type[i]==0)
        {
            p1[i]=lower_bound(a+1,a+1+n,node(r1[i],r2[i]))-a;
            ma[r1[i]]=r2[i];
            update(1,n,1,p1[i],r2[i]);
        }
        if(type[i]==1)
        {   
            p1[i]=lower_bound(a+1,a+1+n,node(r1[i],ma[r1[i]]))-a;
            update(1,n,1,p1[i],0);          
        }
        if(type[i]==2)
        {
            p1[i]=lower_bound(a+1,a+1+n,node(r1[i],ma[r1[i]]))-a;           
            p2[i]=lower_bound(a+1,a+1+n,node(r2[i],ma[r2[i]]))-a;
            if(p1[i]>p2[i])swap(p1[i],p2[i]);
            int t=querym(1,n,1,p1[i],p2[i]);t=pos[t];
            printf("%d\n",get(p1[i])+get(p2[i])-get(t)*2);
        }
    }
    return 0;
}