【BZOJ】3196 Tyvj 1730 二逼平衡树 线段树+平衡树

题目传送门

这题除了烦一点，其实也没什么大不了的嘛……

就是外层一棵区间线段树，内层套上splay，除了第二个操作需要套一个二分，时间复杂度为$O(\log_2^3n)$，其他的操作的时间复杂度都是$O(\log_2^2n)$。

主要是细心吧，耐心一点写都能过的吧。

p.s.话说内存不够导致TLE什么的好鬼啊……还是ZZK大佬最强了，一眼就看出了问题。

附上AC代码：

#include <cstdio>
#include <cctype>
#include <algorithm>
using namespace std;

const int N=5e4+10,M=N*25;
int n,m,x,root[M],size,o,y,v,a[N],ans;
int ch[M][2],f[M],sz[M],w[M];

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

inline void updata(int x){sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1;}

inline void rotate(int x){
    int y=f[x],op=(ch[y][1]==x);
    ch[y][op]=ch[x][op^1];
    if (ch[x][op^1]) f[ch[x][op^1]]=y;
    if (f[x]=f[y]) ch[f[y]][ch[f[y]][1]==y]=x;
    f[y]=x,ch[x][op^1]=y;
    return updata(y),updata(x);
}

inline void splay(int x,int ed,int wz){
    for (int fa=f[x]; fa!=ed; rotate(x),fa=f[x])
        if (f[fa]!=ed) rotate((x==ch[fa][0])==(fa==ch[f[fa]][0])?fa:x);
    if (!ed) root[wz]=x;
    return;
}

inline void ist(int k,int v){
    int now=root[k];
    while (ch[now][w[now]<v]) now=ch[now][w[now]<v];
    w[++size]=v,f[size]=now,ch[size][0]=ch[size][1]=0;
    if (now) ch[now][w[now]<v]=size;
    return splay(size,0,k);
}

inline int get_rank(int k,int v){
    int now=root[k],ret=0;
    while (now)
        if (w[now]<v) ret+=sz[ch[now][0]]+1,now=ch[now][1];
        else now=ch[now][0];
    return ret;
}

inline void del(int k,int v){
    int now=root[k];
    while (w[now]!=v) now=ch[now][w[now]<v];
    splay(now,0,k);
    if (!ch[root[k]][0]) return (void)(f[root[k]=ch[root[k]][1]]=0);
    now=ch[root[k]][0];
    while (ch[now][1]) now=ch[now][1];
    splay(now,root[k],k);
    f[ch[now][1]=ch[root[k]][1]]=now,f[root[k]=now]=0;
    return updata(now);
}

inline void get_pre(int k,int v){
    int now=root[k];
    while (now)
        if (w[now]<v) ans=max(w[now],ans),now=ch[now][1];
        else now=ch[now][0];
}

inline void get_suc(int k,int v){
    int now=root[k];
    while (now)
        if (w[now]>v) ans=min(w[now],ans),now=ch[now][0];
        else now=ch[now][1];
}

//---------- splay ----------

#define lt (k<<1)
#define rt (k<<1|1)
#define mid (l+r>>1)
inline void change(int k,int l,int r,int wz,int v,int o){
    o?ist(k,v): del(k,v);
    if (l==r) return;
    if (wz<=mid) change(lt,l,mid,wz,v,o);
    else change(rt,mid+1,r,wz,v,o);
}

inline int rank(int k,int l,int r,int ql,int qr,int v){
    if (l>qr||r<ql) return 0;
    if (ql<=l&&r<=qr) return get_rank(k,v);
    return rank(lt,l,mid,ql,qr,v)+rank(rt,mid+1,r,ql,qr,v);
}

inline void pre(int k,int l,int r,int ql,int qr,int v){
    if (l>qr||r<ql) return;
    if (ql<=l&&r<=qr) return get_pre(k,v);
    return pre(lt,l,mid,ql,qr,v),pre(rt,mid+1,r,ql,qr,v);
}

inline void suc(int k,int l,int r,int ql,int qr,int v){
    if (l>qr||r<ql) return;
    if (ql<=l&&r<=qr) return get_suc(k,v);
    return suc(lt,l,mid,ql,qr,v),suc(rt,mid+1,r,ql,qr,v);
}

//---------- 线段树 ----------

inline int find(int x,int y,int v){
    int l=0,r=2e9,ret;
    while (l<=r)
        if (rank(1,1,n,x,y,mid)+1<=v) l=mid+1;
        else r=mid-1;
    return r;
}

//---------- 二分 ----------

int main(void){
    read(n),read(m);
    for (int i=1; i<=n; ++i) read(a[i]),change(1,1,n,i,a[i],1);
    while (m--){
        read(o);
        switch (o){
            case 1: read(x),read(y),read(v),printf("%d\n",rank(1,1,n,x,y,v)+1); break;
            case 2: read(x),read(y),read(v),printf("%d\n",find(x,y,v)); break;
            case 3: read(x),read(v); change(1,1,n,x,a[x],0),change(1,1,n,x,a[x]=v,1); break;
            case 4: read(x),read(y),read(v),ans=-(1ll<<31)+1,pre(1,1,n,x,y,v),printf("%d\n",ans); break;
            case 5: read(x),read(y),read(v),ans=(1ll<<31)-1,suc(1,1,n,x,y,v),printf("%d\n",ans); break;
        }
    }
    return 0;
}