P3380 【模板】二逼平衡树（线段树套平衡树）

题目链接

题意：给你n个数的序列，m次操作。操作为各种正常平衡树能做的东西，但是操作区间是l到r，而不是整个区间。

既然是区间操作，那就把线段树套上去就完事。。。

不过一开始写没考虑找不到前驱后继的情况，找bug找了半天。

看学了一手无旋Treap，不用旋转的平衡树是真的香。

无旋Treap原理：

• 首先我们知道Treap由于某个随机数原理，是会使其趋于平衡的
• 当Treap x,y有，x的所有值小于y的所有值，那么我们可以用log的时间把x和y合并
• 我们也可以用log的时间把一棵Treap分裂成小于等于val的部分和大于val的部分，两棵树。
• 然后， 插入，删除，查找可以用若干个分裂合并实现

我写的无旋Treap好像常数大得雅痞。srand(time(NULL));玄学T

无旋Treap板子

namespace Treap{
#define ls(p) T[p].child[0]
#define rs(p) T[p].child[1]
    int tot=0;
    struct node{
        int val,key,sz,child[2];
    }T[maxn*25];
    inline void Push_up(int p){
        T[p].sz=T[ls(p)].sz+T[rs(p)].sz+1;
    }
    inline void Merge(int &p,int x,int y){
        if(!x||!y){
            p=x+y;
            return;
        }
        if(T[x].key<T[y].key)p=x,Merge(rs(p),rs(x),y);
        else p=y,Merge(ls(p),x,ls(y));
        Push_up(p);
    }
    inline void Split(int p,int &x,int &y,int val){
        if(!p){
            x=y=0;
            return;
        }
        if(T[p].val<=val)x=p,Split(rs(p),rs(x),y,val);
        else y=p,Split(ls(p),x,ls(y),val);
        Push_up(p);
    }
    inline void Insert(int &p,int val){
        int x=0,y=0,z=++tot;
        T[z].key=rand();T[z].val=val;T[z].sz=1;
        Split(p,x,y,val);
        Merge(x,x,z);
        Merge(p,x,y);
    }
    inline void Erase(int &p,int val){
        int x=0,y=0,z=0;
        Split(p,x,y,val);
        Split(x,x,z,val-1);
        Merge(z,rs(z),ls(z));
        Merge(x,x,z);
        Merge(p,x,y);
    }
    inline int Kth_num(int p,int k){
        if(k==0)return 0;
        while(T[ls(p)].sz+1!=k){
            if(T[ls(p)].sz>=k)p=ls(p);
            else k-=T[ls(p)].sz+1,p=rs(p);
        }
        return T[p].val;
    }
    inline int Get_rank(int&p,int val){
        int x=0,y=0,res;
        Split(p,x,y,val-1);
        res=T[x].sz;
        Merge(p,x,y);
        return res;
    }
    inline int Pre(int &p,int val){
        int x=0,y=0,res;
        Split(p,x,y,val-1);
        if(T[x].sz)res=Kth_num(x,T[x].sz);
        else res=-INF;
        Merge(p,x,y);
        return res;
    }
    inline int Suf(int &p,int val){
        int x=0,y=0,res;
        Split(p,x,y,val);
        if(T[y].sz)res=Kth_num(y,1);
        else res=INF;
        Merge(p,x,y);
        return res;
    }
}

AC代码

#include<algorithm>
#include<vector>
#include<iostream>
#include<math.h>
#include<cstring>
#include<string>
#include<stack>
#include<map>
#include<set>
#include<unordered_map>
#include<queue>
#include<assert.h>
#include<iomanip>
#include<bitset>
#include<stdio.h>

#define qcin; ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define mp make_pair
#define clr(x) memset(x,0,sizeof x)
#define fmax(x) memset(x,0x3f,sizeof x)
#define finit(x) memset(x,-1,sizeof x)
#define iio(n,m) io(n),io(m)
#define lc(p) (p<<1)
#define rc(p) ((p<<1)|1)

#define dis(l,r) r-l+1
#define gstr(str) scanf("%s",str)
#define glen(str) strlen(str)
using namespace std;

namespace Input{
    const int BUF = 65536;
    char buf[BUF + 1];
    char *head = buf, *tail = buf;
}
inline char inputchar(){
    using namespace Input;
    if(head == tail)
        *(tail = (head = buf) + fread(buf, 1, BUF, stdin)) = 0;
    return *head++;
}
//template<class T>
inline void io(int &ret){
    ret = 0;
    char ch = inputchar();
    while((ch < '0' || ch > '9') && ch != '-')
        ch = inputchar();
    bool neg = false;
    if(ch == '-')
        neg = true, ch = inputchar();
    while(ch >= '0' && ch <= '9')
    {
        ret = ret * 10 + ch - '0';
        ch = inputchar();
    }
    if(neg)
        ret = -ret;
}

typedef long long ll;

typedef pair<int,int>pll;
const int maxn = 100010;
const int mod =  1e9+7;
const ll INF = 2147483647;

typedef int arr[maxn<<2+5];
typedef char str[maxn];
void file(int x){if(x&&fopen("123.in","r")){freopen("123.in","r",stdin);}}

const long double pi = acos(-1);
const double eps=1e-10;
const double delta=0.993;


namespace Treap{
#define ls(p) T[p].child[0]
#define rs(p) T[p].child[1]
    int tot=0;
    struct node{
        int val,key,sz,child[2];
    }T[maxn*25];
    inline void Push_up(int p){
        T[p].sz=T[ls(p)].sz+T[rs(p)].sz+1;
    }
    inline void Merge(int &p,int x,int y){
        if(!x||!y){
            p=x+y;
            return;
        }
        if(T[x].key<T[y].key)p=x,Merge(rs(p),rs(x),y);
        else p=y,Merge(ls(p),x,ls(y));
        Push_up(p);
    }
    inline void Split(int p,int &x,int &y,int val){
        if(!p){
            x=y=0;
            return;
        }
        if(T[p].val<=val)x=p,Split(rs(p),rs(x),y,val);
        else y=p,Split(ls(p),x,ls(y),val);
        Push_up(p);
    }
    inline void Insert(int &p,int val){
        int x=0,y=0,z=++tot;
        T[z].key=rand();T[z].val=val;T[z].sz=1;
        Split(p,x,y,val);
        Merge(x,x,z);
        Merge(p,x,y);
    }
    inline void Erase(int &p,int val){
        int x=0,y=0,z=0;
        Split(p,x,y,val);
        Split(x,x,z,val-1);
        Merge(z,rs(z),ls(z));
        Merge(x,x,z);
        Merge(p,x,y);
    }
    inline int Kth_num(int p,int k){
        if(k==0)return 0;
        while(T[ls(p)].sz+1!=k){
            if(T[ls(p)].sz>=k)p=ls(p);
            else k-=T[ls(p)].sz+1,p=rs(p);
        }
        return T[p].val;
    }
    inline int Get_rank(int&p,int val){
        int x=0,y=0,res;
        Split(p,x,y,val-1);
        res=T[x].sz;
        Merge(p,x,y);
        return res;
    }
    inline int Pre(int &p,int val){
        int x=0,y=0,res;
        Split(p,x,y,val-1);
        if(T[x].sz)res=Kth_num(x,T[x].sz);
        else res=-INF;
        Merge(p,x,y);
        return res;
    }
    inline int Suf(int &p,int val){
        int x=0,y=0,res;
        Split(p,x,y,val);
        if(T[y].sz)res=Kth_num(y,1);
        else res=INF;
        Merge(p,x,y);
        return res;
    }
}

arr T;

inline void add(int p,int l,int r,int pos,int val){
    Treap::Insert(T[p],val);
    if(l==r)return;
    int md=(l+r)>>1;
    if(pos<=md)add(lc(p),l,md,pos,val);
    if(pos>md)add(rc(p),md+1,r,pos,val);
}
inline void del(int p,int l,int r,int pos,int val){
    Treap::Erase(T[p],val);
    if(l==r)return;
    int md=(l+r)>>1;
    if(pos<=md)del(lc(p),l,md,pos,val);
    if(pos>md)del(rc(p),md+1,r,pos,val);
}
inline int qry_nth(int p,int l,int r,int  ql,int qr,int val){
    if(ql<=l&&r<=qr)return Treap::Get_rank(T[p],val);
    int md=(l+r)>>1,res=0;
   // cout<<l<<' '<<r<<endl;
    if(ql<=md)res+=qry_nth(lc(p),l,md,ql,qr,val);
    if(qr>md)res+=qry_nth(rc(p),md+1,r,ql,qr,val);
    return res;
}
inline int qry_pre(int p,int l,int r,int  ql,int qr,int val){
    if(ql<=l&&r<=qr)return Treap::Pre(T[p],val);
    int md=(l+r)>>1,res=-INF;
    if(ql<=md)res=max(res,qry_pre(lc(p),l,md,ql,qr,val));
    if(qr>md)res=max(res,qry_pre(rc(p),md+1,r,ql,qr,val));
    return res;
}
inline int qry_suf(int p,int l,int r,int  ql,int qr,int val){
    if(ql<=l&&r<=qr)return Treap::Suf(T[p],val);
    int md=(l+r)>>1,res=INF;
    if(ql<=md)res=min(res,qry_suf(lc(p),l,md,ql,qr,val));
    if(qr>md)res=min(res,qry_suf(rc(p),md+1,r,ql,qr,val));
    return res;
}
int n,m,op,x,y,z;
arr a;
int main(){
    file(1);
    srand(19260817);//srand(rand());
    iio(n,m);
    for(int i=1;i<=n;i++){
        io(a[i]);
        add(1,1,n,i,a[i]);
    }
    while(m--){
        io(op),iio(x,y);
        if(op!=3)io(z);
        if(op==1)printf("%d\n",qry_nth(1,1,n,x,y,z)+1);
        if(op==2){
            int L=0,R=1e8;
            while(L<R){
                int md=(L+R+1)>>1;
                if(qry_nth(1,1,n,x,y,md)+1<=z)L=md;
                else R=md-1;
            }
            printf("%d\n",R);
        }
        if(op==3){
            del(1,1,n,x,a[x]);
            a[x]=y;
            add(1,1,n,x,a[x]);
        }
        if(op==4)printf("%d\n",qry_pre(1,1,n,x,y,z));
        if(op==5)printf("%d\n",qry_suf(1,1,n,x,y,z));
    }
}