[BZOJ3196]二逼平衡树（线段树套splay）

题目：

我是超链接

题解：

卡常小王子终于卡进了BZOJ的时限（其实只是空间开的不对啊）
线段树套splay，简单地说就是线段树的每一个节点都吊着一颗splay，表示的是线段树当前节点所表示的区间的点，按权值排序。
Q1：线段树常规查询区间，每一次统计小于k的点的个数再相加。
Q2：二分答案，每二分到一个答案查询一下这个答案在这个区间内的排名。
Q3：相当于线段树的点修改，在splay中删除再插入即可。
Q4：线段树常规查询区间，每一次找区间内比k小的最大的数，然后取max
Q5：类似于Q4，每一次找区间内比k大的最小的数，然后取min

代码：

#include <cstdio>
using namespace std;
const int N=3000005;
const int INF=2147483647;
int ch[N][2],size[N],cnt[N],f[N],root[N],sz,key[N],maxx,minn,Max,Min,a[N];
int get(int x){return ch[f[x]][1]==x;}
int max(int a,int b){if (a>b) return a;return b;}
int min(int a,int b){if (a>b) return b;return a;}
int read()
{
    int x=0, f=1;char ch = getchar();
    while (ch < '0' || ch > '9') { if (ch == '-') f=-1; ch = getchar(); }
    while (ch >= '0' && ch <= '9') { x = x * 10 + ch - '0';ch = getchar(); }
    return x * f;
}
void updata(int now)
{
    size[now]=cnt[now];
    if (ch[now][0]) size[now]+=size[ch[now][0]];
    if (ch[now][1]) size[now]+=size[ch[now][1]];
}
void rotate(int x)
{
    int old=f[x],oldf=f[old],which=get(x);
    f[x]=oldf; if (oldf) ch[oldf][get(old)]=x;
    ch[old][which]=ch[x][which^1]; f[ch[x][which^1]]=old;
    ch[x][which^1]=old; f[old]=x;
    updata(old); updata(x);
}
void splay(int now,int x)
{
    for (;f[x];rotate(x))
      if (f[f[x]]) rotate(get(x)==get(f[x])?f[x]:x);
    root[now]=x;
}
int pre(int i)
{
    int now=ch[root[i]][0];
    while (ch[now][1]) now=ch[now][1];
    return now; 
}
void insert(int i,int x)
{
    int now=root[i];
    if (!now) 
    {
        ++sz; key[sz]=x; size[sz]=cnt[sz]=1;
        root[i]=sz;return; 
    }
    int fa=0;
    while (1)
    {
        if (key[now]==x)
        {
            cnt[now]++; updata(now);
            updata(fa); splay(i,now);
            return;
        }
        fa=now;
        now=ch[now][key[now]<x];
        if (!now)
        {
            ++sz;key[sz]=x;size[sz]=cnt[sz]=1;
            f[sz]=fa; ch[fa][key[fa]<x]=sz;
            updata(sz); updata(fa); splay(i,sz);
            return;
        }
    }
}
int find(int i,int x)//值为x的now值
{
    int now=root[i];
    while (1)
    {
        if (key[now]==x) {splay(i,now); return now;}
        now=ch[now][key[now]<x];
    }
} 
int findrank(int i,int x)//值为x的排名 
{
    int now=root[i],ans=0;
    while (1)
    {
        if (!now) return ans;
        if (key[now]==x) return ans+size[ch[now][0]];
        if (key[now]<x)
        {
            ans+=size[ch[now][0]]+cnt[now];
            now=ch[now][1];
        }else now=ch[now][0];
    }
}
void del(int i,int x)
{
    int now=find(i,x);
    if (cnt[now]>1) {cnt[now]--;return;}
    if (!ch[now][0] && !ch[now][1]) {root[i]=0;return;}
    if (!ch[now][0]){f[ch[now][1]]=0; root[i]=ch[now][1]; splay(i,ch[now][1]);  return; }
    else if (!ch[now][1]){f[ch[now][0]]=0; root[i]=ch[now][0]; splay(i,ch[now][0]); return;}
    int leftbig=pre(i);
    splay(i,leftbig);
    f[leftbig]=0; f[ch[now][1]]=leftbig;
    ch[leftbig][1]=ch[now][1]; updata(leftbig);
    return;
}
void seginsert(int now,int l,int r,int x,int v)
{
    insert(now,v);  if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) seginsert(now<<1,l,mid,x,v);
    else seginsert(now<<1|1,mid+1,r,x,v);
}
void segdel(int now,int l,int r,int x,int v)
{
    del(now,v); if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) segdel(now<<1,l,mid,x,v);
    else segdel(now<<1|1,mid+1,r,x,v);
}
int segrank(int now,int l,int r,int lrange,int rrange,int v)
{
    if (lrange<=l && rrange>=r) return findrank(now,v);
    int mid=(l+r)>>1,ans=0;
    if (lrange<=mid) ans+=segrank(now<<1,l,mid,lrange,rrange,v);
    if (rrange>mid) ans+=segrank(now<<1|1,mid+1,r,lrange,rrange,v);
    return ans;
}
void change(int now,int l,int r,int x,int v,int vv)
{
    del(now,v); insert(now,vv); 
    if (l==r) return;
    int mid=(l+r)>>1;
    if (x<=mid) change(now<<1,l,mid,x,v,vv);
    else change(now<<1|1,mid+1,r,x,v,vv);
}
void findmax(int i,int x)
{
    int now=root[i];
    while (now)
    {
        if (key[now]<x) maxx=max(maxx,key[now]),now=ch[now][1];
        else now=ch[now][0];
    }
} 
void qurrymax(int now,int l,int r,int lrange,int rrange,int v)
{
    if (lrange<=l && rrange>=r){findmax(now,v);return;}
    int mid=(l+r)>>1;
    if (lrange<=mid) qurrymax(now<<1,l,mid,lrange,rrange,v);
    if (rrange>mid) qurrymax(now<<1|1,mid+1,r,lrange,rrange,v);
}
void findmin(int i,int x)
{
    int now=root[i];
    while (now)
    {
        if (key[now]>x) minn=min(minn,key[now]),now=ch[now][0];
        else now=ch[now][1];
    }
} 
void qurrymin(int now,int l,int r,int lrange,int rrange,int v)
{
    if (lrange<=l && rrange>=r){findmin(now,v);return;}
    int mid=(l+r)>>1;
    if (lrange<=mid) qurrymin(now<<1,l,mid,lrange,rrange,v);
    if (rrange>mid) qurrymin(now<<1|1,mid+1,r,lrange,rrange,v);
}
int main()
{
    int n,m;n=read();m=read(); Min=INF; Max=-INF;
    for (int i=1;i<=n;i++) a[i]=read(),seginsert(1,1,n,i,a[i]),Min=min(Min,a[i]),Max=max(Max,a[i]);
    while (m--)
    {
        int id,x,y,k;id=read();
        switch(id)
        {
            case 1:x=read();y=read();k=read();printf("%d\n",segrank(1,1,n,x,y,k)+1);break;
            case 2:{
                x=read();y=read();k=read();
                int l=Min,r=Max;
                while (l<=r)
                {
                    int mid=(l+r)>>1;
                    if (segrank(1,1,n,x,y,mid)<k) l=mid+1;
                    else r=mid-1;
                }
                printf("%d\n",r);
                break;
            }
            case 3:x=read();k=read();change(1,1,n,x,a[x],k);a[x]=k;Max=max(Max,k);Min=min(Min,k);break;
            case 4:maxx=-INF;x=read();y=read();k=read();qurrymax(1,1,n,x,y,k);printf("%d\n",maxx);break;
            case 5:minn=INF;x=read();y=read();k=read();qurrymin(1,1,n,x,y,k);printf("%d\n",minn);break; 
        }
    }
}