bzoj 3196 Tyvj 1730 二逼平衡树

Posted 2020-11-20 narh

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3196

第一道的树套树。

线段树套平衡树。找了一个当模板。感觉还好啦。但跑得很慢。卡时过了。

1.好神奇呀，竟然都要加-INF和INF。别忘了求siz时-1；

2.去右儿子找的时候别忘了减去左儿子和自己的siz；

3.凡是有splay的地方就别忘了 & ；

4.想要 < 的就把它放在return y?y：cr里，想要 <= 的也就把它放在那个地方，大约就是这样就行；

5.线段树的 tt 别和Splay的 tot 混用，不然线段树节点的数组就得……

6.2e6是怎么算的呀？1e6为什么不行？

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=5e4+5,M=2e6+5,INF=1e9;
int n,m,t[N],tot,tt,fa[M],c[M][2],val[M],siz[M];//
struct Node{
    int l,r,ls,rs,rt;
}a[N<<1];
void updt(int cr)
{
    siz[cr]=siz[c[cr][0]]+siz[c[cr][1]]+1;
}
void rotate(int x,int &rt)
{
    int y=fa[x],z=fa[y],d=(x==c[y][1]);
    if(y==rt)rt=x;
    else c[z][y==c[z][1]]=x;
    fa[y]=x;fa[x]=z;fa[c[x][!d]]=y;
    c[y][d]=c[x][!d];c[x][!d]=y;
    updt(y);updt(x);
}
void splay(int x,int &rt)
{
    while(x!=rt)
    {
        int y=fa[x],z=fa[y];
        if(y!=rt)
        {
            if((x==c[y][0])^(y==c[z][0]))rotate(x,rt);
            else rotate(y,rt);
        }
        rotate(x,rt);
    }
}
int find_k(int cr,int x)
{
    if(siz[c[cr][0]]==x-1)return cr;
    if(siz[c[cr][0]]>=x)return find_k(c[cr][0],x);
    return find_k(c[cr][1],x-c[cr][0]-1);
}
int find_pre(int cr,int x)
{
    if(!cr)return 0;
    if(val[cr]>=x)return find_pre(c[cr][0],x);
    int y=find_pre(c[cr][1],x);
    return y?y:cr;
}
void insert(int &rt,int x)
{
    int y=find_pre(rt,x);splay(y,rt);
    y=find_k(c[rt][1],1);splay(y,c[rt][1]);
    fa[++tot]=y;c[y][0]=tot;siz[tot]=1;val[tot]=x;
    updt(y);updt(rt);
}
void del(int &rt,int x)
{
    int y=find_pre(rt,x);splay(y,rt);
    y=find_k(c[rt][1],2);splay(y,c[rt][1]);
    fa[c[y][0]]=0;c[y][0]=0;
    updt(y);updt(rt);
}
void build(int l,int r,int cr)
{
    a[cr].l=l;a[cr].r=r;a[cr].rt=++tot;
    fa[tot]=0;c[tot][1]=tot+1;c[tot][0]=0;val[tot]=-INF;siz[tot]=2;
    fa[++tot]=tot-1;c[tot][1]=c[tot][0]=0;val[tot]=INF;siz[tot]=1;
    for(int i=l;i<=r;i++)insert(a[cr].rt,t[i]);
    if(l==r)return;
    int mid=((l+r)>>1);
    a[cr].ls=++tt;build(l,mid,tt);
    a[cr].rs=++tt;build(mid+1,r,tt);
}
int find_Suc(int rt,int x)    //Suc表示找的不是正常后继(>=)
{
    if(!rt)return 0;
    if(val[rt]<x)return find_Suc(c[rt][1],x);
    int y=find_Suc(c[rt][0],x);
    return y?y:rt;
}
int find_suc(int rt,int x)
{
    if(!rt)return 0;
    if(val[rt]<=x)return find_suc(c[rt][1],x);
    int y=find_suc(c[rt][0],x);
    return y?y:rt;
}
int query_rnk(int &rt,int x)    //&
{
    int y=find_Suc(rt,x);splay(y,rt);
    return siz[c[rt][0]]-1;    //-INF
}
int query_Rnk(int &rt,int x)
{
    int y=find_suc(rt,x);splay(y,rt);
    return siz[c[rt][0]]-1;
}
int query_rnk(int L,int R,int cr,int x)
{
    if(a[cr].l>=L&&a[cr].r<=R)return query_rnk(a[cr].rt,x);
    int mid=((a[cr].l+a[cr].r)>>1),ret=0;
    if(mid>=L)ret+=query_rnk(L,R,a[cr].ls,x);
    if(mid<R)ret+=query_rnk(L,R,a[cr].rs,x);
    return ret;
}
int query_Rnk(int L,int R,int cr,int x)
{
    if(a[cr].l>=L&&a[cr].r<=R)return query_Rnk(a[cr].rt,x);
    int mid=((a[cr].l+a[cr].r)>>1),ret=0;
    if(mid>=L)ret+=query_Rnk(L,R,a[cr].ls,x);
    if(mid<R)ret+=query_Rnk(L,R,a[cr].rs,x);
    return ret;
}
void mdfy(int P,int cr,int x)
{
    insert(a[cr].rt,x);del(a[cr].rt,t[P]);
    if(a[cr].l==a[cr].r)return;
    int mid=((a[cr].l+a[cr].r)>>1);
    if(P<=mid)mdfy(P,a[cr].ls,x);
    else mdfy(P,a[cr].rs,x);
}
int query_pre(int L,int R,int cr,int x)
{
    if(a[cr].l>=L&&a[cr].r<=R)return val[find_pre(a[cr].rt,x)];
    int ret=-INF,mid=((a[cr].l+a[cr].r)>>1);
    if(mid>=L)ret=max(ret,query_pre(L,R,a[cr].ls,x));
    if(mid<R)ret=max(ret,query_pre(L,R,a[cr].rs,x));
    return ret;
}
int query_suc(int L,int R,int cr,int x)
{
    if(a[cr].l>=L&&a[cr].r<=R)return val[find_suc(a[cr].rt,x)];
    int ret=INF,mid=((a[cr].l+a[cr].r)>>1);
    if(mid>=L)ret=min(ret,query_suc(L,R,a[cr].ls,x));
    if(mid<R)ret=min(ret,query_suc(L,R,a[cr].rs,x));
    return ret;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&t[i]);
    tt=1;build(1,n,1);
    int op,x,y,k;
    while(m--)
    {
        scanf("%d%d%d",&op,&x,&y);if(op!=3)scanf("%d",&k);
        if(op==1)printf("%d
",query_rnk(x,y,1,k)+1);
        if(op==2){
            int l=0,r=INF,ans=0;
            while(l<=r)
            {
                int mid=((l+r)>>1);
                if(query_Rnk(x,y,1,mid)>=k)ans=mid,r=mid-1;//不+1：包含了全部==自己的 
                else l=mid+1;
            }
            printf("%d
",ans);
        }
        if(op==3)mdfy(x,1,y),t[x]=y;
        if(op==4)printf("%d
",query_pre(x,y,1,k));
        if(op==5)printf("%d
",query_suc(x,y,1,k));
    }
    return 0;
}