bzoj3196 二逼平衡树

题目链接

平衡树系列最后一题

坑啊

10s时间限制跑了9764ms。。。还是要学一学bit套主席树啦。。。

经典的线段树套treap。。。至于第一发为什么要TLE（我不会告诉你treap插入的时候忘了旋转 WOC）

自认为treap写的挺好看的（欢迎来喷）

 #include<algorithm>
 #include<iostream>
 #include<cstdlib>
 #include<cstring>
 #include<cstdio>
 #include<string>
 #include<cmath>
 #include<ctime>
 #include<queue>
 #include<stack>
 #include<map>
 #include<set>
 #define rre(i,r,l) for(int i=(r);i>=(l);i--)
 #define re(i,l,r) for(int i=(l);i<=(r);i++)
 #define Clear(a,b) memset(a,b,sizeof(a))
 #define inout(x) printf("%d",(x))
 #define douin(x) scanf("%lf",&x)
 #define strin(x) scanf("%s",(x))
 #define LLin(x) scanf("%lld",&x)
 #define op operator
 #define CSC main
 typedef unsigned long long ULL;
 typedef const int cint;
 typedef long long LL;
 using namespace std;
 cint inf=;
 void inin(int &ret)
 {
     ret=;int f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=;ch=getchar();}
     while(ch>=''&&ch<='')ret*=,ret+=ch-'',ch=getchar();
     ret=f?-ret:ret;
 }
 int ch[][],c[],s[],rr[],w[],ed;
 struct segtree
 {
         int l,r,root;
         segtree(){root=;}
         void maintain(int k){if(k)s[k]=c[k]+s[ch[k][]]+s[ch[k][]];}
         void rotate(int &k,int d)
         {
             int p=ch[k][d^];
             ch[k][d^]=ch[p][d];
             ch[p][d]=k;s[p]=s[k];
             maintain(k),k=p;
         }
     private:
         void add(int &k,const int &x)
         {
             if(!k)
             {
                 k=++ed,w[k]=x,rr[k]=rand();s[k]=c[k]=;ch[k][]=ch[k][]=;
                 return ;
             }s[k]++;
             if(x==w[k]){c[k]++;return ;}
             int d=x>w[k];add(ch[k][d],x);
             if(rr[ch[k][d]]<rr[k])rotate(k,d^);
         }
         bool del(int &k,const int &x)
         {
             if(!k)return ;
             if(x==w[k])
             {
                 if(c[k]>){c[k]--,s[k]--;return ;}
                 if(!ch[k][]){k=ch[k][];return ;}
                 if(!ch[k][]){k=ch[k][];return ;}
                 if(rr[ch[k][]]<rr[ch[k][]])rotate(k,);
                 else rotate(k,);
                 return del(k,x);
             }
             int d=x>w[k];
             if(del(ch[k][d],x))
             {
                 s[k]--;
                 return ;
             }
             else return ;
         }
         int findrank(int k,const int &x)
         {
             if(!k)return ;
             int pp=(ch[k][]?s[ch[k][]]:);
             if(x==w[k])return pp;
             if(x>w[k])return pp+c[k]+findrank(ch[k][],x);
             return findrank(ch[k][],x);
         }
         int findwei(int k,const int &x)
         {
             if(!k)return ;
             int pp=(ch[k][]?s[ch[k][]]:);
             if(x<=pp)return findwei(ch[k][],x);
             if(x>pp+c[k])return findwei(ch[k][],x-pp-c[k]);
             return w[k];

         }
         void findqian(int k,const int &x,int &ans)
         {
             if(!k)return ;
             if(x>w[k])ans=w[k],findqian(ch[k][],x,ans);
             else findqian(ch[k][],x,ans);
         }
         void findhou(int k,const int &x,int &ans)
         {
             if(!k)return ;
             if(x<w[k])ans=w[k],findhou(ch[k][],x,ans);
             else findhou(ch[k][],x,ans);
         }
     public:
         void add(int x){add(root,x);}
         void del(int x){del(root,x);}
         int findrank(int x){return findrank(root,x);}
         int findwei(int x){return findwei(root,x);}
         int findqian(int x)
         {
             int ans=;findqian(root,x,ans);
             return ans;
         }
         int findhou(int x)
         {
             int ans=;findhou(root,x,ans);
             return !ans?inf:ans;
         }
 }t[];
 int qi[];
 void build(int k,int l,int r,int x)
 {
     t[k].l=l,t[k].r=r,t[k].add(qi[x]);
     if(l==r)return ;int mid=(l+r)>>;
     if(x<=mid)build(k<<,l,mid,x);
     else build(k<<|,mid+,r,x);
 }
 int findrank(int k,int l,int r,int x)
 {
     if(t[k].l>=l&&t[k].r<=r)return t[k].findrank(x);
     int mid=(t[k].l+t[k].r)>>,p1=k<<,p2=p1|;
     if(r<=mid)return findrank(p1,l,r,x);
     if(l>mid)return findrank(p2,l,r,x);
     return findrank(p1,l,r,x)+findrank(p2,l,r,x);
 }
 int findqian(int k,int l,int r,int x)
 {
     if(t[k].l>=l&&t[k].r<=r)return t[k].findqian(x);
     int mid=(t[k].l+t[k].r)>>,p1=k<<,p2=p1|;
     if(r<=mid)return findqian(p1,l,r,x);
     if(l>mid)return findqian(p2,l,r,x);
     return max(findqian(p1,l,r,x),findqian(p2,l,r,x));
 }
 int findhou(int k,int l,int r,int x)
 {
     if(t[k].l>=l&&t[k].r<=r)return t[k].findhou(x);
     int mid=(t[k].l+t[k].r)>>,p1=k<<,p2=p1|;
     if(r<=mid)return findhou(p1,l,r,x);
     if(l>mid)return findhou(p2,l,r,x);
     return min(findhou(p1,l,r,x),findhou(p2,l,r,x));
 }
 int findwei(int l,int r,int x)
 {
     int ll=,rr=inf,mid,ans;
     while(ll<=rr)
     {
         mid=(ll+rr)>>;
         int rank=findrank(,l,r,mid)+;
         if(rank<=x)ans=mid,ll=mid+;else rr=mid-;
     }
     return findqian(,l,r,ans+);
 }
 void change(int k,int x,int w)
 {
     t[k].del(qi[x]),t[k].add(w);
     if(t[k].l==t[k].r)return ;
     int mid=(t[k].l+t[k].r)>>;
     if(x<=mid)change(k<<,x,w);
     else change(k<<|,x,w);
 }
 int n,m;
 int CSC()
 {
     freopen("in.in","r",stdin);
     freopen("out.out","w",stdout);
     inin(n),inin(m);
     re(i,,n)inin(qi[i]);
     re(i,,n)build(,,n,i);
     re(i,,m)
     {
         int opt,q,ww,e;
         inin(opt);
         if(opt==)
         {
             inin(q),inin(ww),inin(e);
             printf("%d\n",findrank(,q,ww,e)+);
         }
         else if(opt==)
         {
             inin(q),inin(ww),inin(e);
             printf("%d\n",findwei(q,ww,e));
         }
         else if(opt==)
         {
             inin(q),inin(ww);
             change(,q,ww);qi[q]=ww;
         }
         else if(opt==)
         {
             inin(q),inin(ww),inin(e);
             printf("%d\n",findqian(,q,ww,e));
         }
         else
         {
             inin(q),inin(ww),inin(e);int ret=findhou(,q,ww,e);
 //            printf("%d\n",ret==inf?0:ret);
             printf("%d\n",ret);
         }
     }
     return ;
 }