树套树-线段树套平衡树

树套树-线段树套平衡树

这个代码写起来真心累。。。

今天反正是不想再写一遍了

虽然好像十分无脑的样子。。。

思路十分简单，实现十分自然。。。

bzoj 3196 二逼平衡树

#include<stdio.h>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<vector>
#include<map>
#include<set>
#include<cmath>
#include<iostream>
#include<queue>
#include<string>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
typedef long double ld;
typedef unsigned long long ull;
typedef pair<long long,long long> pll;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define rep(i,j,k)  for(register int i=(int)(j);i<=(int)(k);i++)
#define rrep(i,j,k) for(register int i=(int)(j);i>=(int)(k);i--)

ll read(){
    ll x=0,f=1;char c=getchar();
    while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0' && c<='9'){x=x*10+c-'0';c=getchar();}
    return x*f;
}

#define lc (i<<1)
#define rc (i<<1|1)
const int maxn=50050;
struct Node{
    int l,r,root;
} tr[maxn*3];
int n,m,a[maxn],b[maxn],tot;
struct TreapNode{
    int ls,rs,val,rnd,sz;
} treap[maxn*20];
int sta[maxn],top;

int newnode(int v){
    treap[++tot].val=v;treap[tot].rnd=rand();
    treap[tot].sz=1;treap[tot].ls=treap[tot].rs=0;
    return tot;
}
void update(int i){
    treap[i].sz=treap[treap[i].ls].sz+treap[treap[i].rs].sz+1;
}
void build(int i,int l,int r){
    tr[i].l=l,tr[i].r=r;
    if(l==r){
        tr[i].root=newnode(a[l]);
        return;
    }
    rep(k,l,r) b[k]=a[k];
    sort(b+l,b+r+1);
    int nw=0,pre=0;top=0;
    rep(k,l,r){
        nw=newnode(b[k]);pre=0;
        while(top && treap[sta[top]].rnd>treap[nw].rnd){
            update(sta[top]);
            pre=sta[top];
            sta[top--]=0;
        }
        if(top) treap[sta[top]].rs=nw;
        treap[nw].ls=pre;sta[++top]=nw;
    }
    while(top) update(sta[top--]);
    tr[i].root=sta[1];
    int md=(l+r)>>1;
    build(lc,l,md);build(rc,md+1,r);
}
void split(int &x,int &y,int val,int nw){
    if(nw==0){x=y=0;return;}
    if(treap[nw].val<=val){
        x=nw;split(treap[x].rs,y,val,treap[x].rs);
    }
    else{
        y=nw;split(x,treap[y].ls,val,treap[y].ls);
    }
    update(nw);
}
int merge(int x,int y){
    if(!x || !y) return x+y;
    int nw;
    if(treap[x].rnd<treap[y].rnd){
        treap[x].rs=merge(treap[x].rs,y);
        update(x);return x;
    }
    else{
        treap[y].ls=merge(x,treap[y].ls);
        update(y);return y;
    }
}
int fnd1(int i,int v){
    int x,y;
    split(x,y,v-1,tr[i].root);
    int ret=treap[x].sz;
    //cerr<<treap[x].val<<' '<<treap[treap[x].ls].val<<' '<<treap[x].ls<<' '<<treap[x].rs<<endl;
    tr[i].root=merge(x,y);return ret;
}
int ask1(int i,int l,int r,int L,int R,int v){
    if(l>=L && r<=R){
        int ret=fnd1(i,v);
    //  cerr<<l<<' '<<r<<' '<<ret<<endl;
        return ret;
    }
    if(l>R || r<L) return 0;
    int md=(l+r)>>1;
    return ask1(lc,l,md,L,R,v)+ask1(rc,md+1,r,L,R,v);
}
int query1(int l,int r,int k){
    int ret=ask1(1,1,n,l,r,k)+1;
    //cerr<<l<<' '<<r<<' '<<k<<' '<<ret<<endl;
    return ret;
}
int query2(int l,int r,int k){
    int L=0,R=100000000;
    while(L+1<R){
        int md=(L+R)>>1;
        int num=query1(l,r,md);
        if(num<=k) L=md;
        else R=md-1;
    //  cout<<md<<"is"<<num<<endl;
    }
    int num=query1(l,r,L+1);
    if(num<=k) return L+1;
    else return L;
}
void modify(int i,int l,int r,int pos,int v1,int v2){
    int x,y,z;
    split(x,z,v2,tr[i].root);
    split(x,y,v2-1,x);
    y=merge(treap[y].ls,treap[y].rs);
    tr[i].root=merge(merge(x,y),z);
    split(x,y,v1,tr[i].root);
    tr[i].root=merge(merge(x,newnode(v1)),y);
    if(l==r) return;
    int md=(l+r)>>1;
    if(pos<=md) modify(lc,l,md,pos,v1,v2);
    else modify(rc,md+1,r,pos,v1,v2);
}
void change(int pos,int v1,int v2){
    modify(1,1,n,pos,v1,v2);
}
int fnd4(int i,int v){
    int nw=tr[i].root,x,y;
    split(x,y,v-1,nw);
    int nod=x;
    while(treap[nod].rs) nod=treap[nod].rs;
    int ret=treap[nod].val;
    tr[i].root=merge(x,y);
    return ret;
}
int ask4(int i,int l,int r,int L,int R,int v){
    if(l>=L && r<=R) return fnd4(i,v);
    if(l>R || r<L) return -1e9;
    int md=(l+r)>>1;
    return max(ask4(lc,l,md,L,R,v),ask4(rc,md+1,r,L,R,v));
}
int query4(int l,int r,int k){
    return ask4(1,1,n,l,r,k);
}
int fnd5(int i,int v){
    int nw=tr[i].root,x,y;
    split(x,y,v,nw);
    int nod=y;
    while(treap[nod].ls) nod=treap[nod].ls;
    int ret=treap[nod].val;if(nod==0) ret=1e9;
    tr[i].root=merge(x,y);
    return ret;
}
int ask5(int i,int l,int r,int L,int R,int v){
    if(l>=L && r<=R) return fnd5(i,v);
    if(l>R || r<L) return 1e9;
    int md=(l+r)>>1;
    return min(ask5(lc,l,md,L,R,v),ask5(rc,md+1,r,L,R,v));
}
int query5(int l,int r,int k){
    return ask5(1,1,n,l,r,k);
}
int main(){
    srand(20020709);
    //freopen("in","r",stdin);
    n=read(),m=read();
    rep(i,1,n) a[i]=read();
    build(1,1,n);
    while(m--){
        int op=read();
        if(op==1){
            int l=read(),r=read(),k=read();
            printf("%d\n",query1(l,r,k));
        }
        else if(op==2){
            int l=read(),r=read(),k=read();
            printf("%d\n",query2(l,r,k));
        }
        else if(op==3){
            int pos=read(),v=read();
            change(pos,v,a[pos]);
            a[pos]=v;
        }
        else if(op==4){
            int l=read(),r=read(),k=read();
            printf("%d\n",query4(l,r,k));
        }
        else{
            int l=read(),r=read(),k=read();
            printf("%d\n",query5(l,r,k));
        }
    }
    return 0;
}

/*
9 6
4 2 2 1 9 4 0 1 1
2 1 4 3
3 4 10
2 1 4 3
1 2 5 9
4 3 9 5
5 2 8 5
*/