Zoj 2112 线段树套Treap

树套树入门题
树套树空间复杂度分析:维护一个长度为n的序列,线段树的结点数一般为4n,而Treap则是对于每一个元素(包括重复出现的)都需要一个空间,那么第1层的所需要的空间为(1*n),第二层为(2*n/2)……所以其实为O(h)*n.
那么所需的空间为O(nlogn)但在实际运用中,所形成的线段树往往不是完全二叉树,即最后一层并不满,所以上述空间会非常吃紧,所以一般为O(4nlog4n)即结点数的log倍
PS:大常数选手跑了4000ms

#include<iostream>#include<cstdio>#include<cmath>#include<cstring>#include<algorithm>#define fo(i,a,b) for(int i=a;i<=b;i++)#define fod(i,a,b) for(int i=a;i>=b;i--)using namespace std;const int N=200000+10,M=1300001;int l[M],r[M],s[M],v[M],w[M],rnd[M],a[N],n,m,sz,root[N],tmp;void update(int k){s[k]=s[l[k]]+s[r[k]]+w[k];}void rturn(int &k){int t=l[k];l[k]=r[t];r[t]=k;update(k);update(t);k=t;}void lturn(int &k){int t=r[k];r[k]=l[t];l[t]=k;update(k);update(t);k=t;}void insert(int &k,int val){    if(!k){k=++sz;l[k]=r[k]=0;w[k]=s[k]=1;v[k]=val;rnd[k]=rand();return ;}    s[k]++;    if(val==v[k]) w[k]++;    else if(val>v[k]) {insert(r[k],val);if(rnd[r[k]]<rnd[k])lturn(k);}         else         {insert(l[k],val);if(rnd[l[k]]<rnd[k])rturn(k);}}void del(int &k,int val){    if(!k)return;    if(val==v[k]) {        if(w[k]>1){w[k]--;s[k]--;return;}        if(l[k]*r[k]==0)k=l[k]+r[k];        else if(rnd[r[k]]<rnd[l[k]])lturn(k),del(k,val);        else                        rturn(k),del(k,val);    }else if(val>v[k])         s[k]--,del(r[k],val);   else s[k]--,del(l[k],val); }void find(int k,int val){    if(!k)return;    if(val<v[k])find(l[k],val);    else {tmp+=(s[l[k]]+w[k]);find(r[k],val);}}void build(int k,int l,int r,int pos,int val){    insert(root[k],val);    if(l==r)return;    int mid=(l+r)>>1;    if(pos<=mid)build(k<<1,l,mid,pos,val);    else build(k<<1|1,mid+1,r,pos,val);}void change(int k,int l,int r,int pos,int val,int Del){    del(root[k],Del);    insert(root[k],val);    if(l==r)return;    int mid=(l+r)>>1;    if(pos<=mid)change(k<<1,l,mid,pos,val,Del);    else change(k<<1|1,mid+1,r,pos,val,Del);}void que(int k,int l,int r,int ql,int qr,int val){    if(ql==l&&r==qr){find(root[k],val);return;}    int mid=(l+r)>>1;    if(qr<=mid)que(k<<1,l,mid,ql,qr,val);    else if(ql>mid)que(k<<1|1,mid+1,r,ql,qr,val);    else {        que(k<<1,l,mid,ql,mid,val);        que(k<<1|1,mid+1,r,mid+1,qr,val);    }}int main(){    int T;scanf("%d",&T);    while(T--) {        memset(root,0,sizeof(root));        sz=0;        scanf("%d%d",&n,&m);        fo(i,1,n) scanf("%d",&a[i]);        fo(i,1,n) build(1,1,n,i,a[i]);        for(int l,r,x,i=1;i<=m;i++) {            char ch[3];scanf("%s",ch);            if(ch[0]=='C') {                scanf("%d%d",&l,&r);                change(1,1,n,l,r,a[l]);a[l]=r;            }else {                scanf("%d%d%d",&l,&r,&x);                int L=0,R=1e9;                while(L<=R){                    int mid=(L+R)>>1;                    tmp=0;que(1,1,n,l,r,mid);                    if(tmp>=x)R=mid-1;                    else L=mid+1;                }                printf("%d\n",L);            }        }    }    return 0;}