【BZOJ3196】二逼平衡树（树状数组，线段树）

题面

BZOJ题面

题解

如果不存在区间修改操作：
搞一个权值线段树
区间第K大—>直接在线段树上二分
某个数第几大—>查询一下区间的size和
某个数的前缀—>先查一下他是区间第几大，再求他-1大
某个数的后缀—>和上面那个有区别吗？？？

现在有了区间修改操作
多搞一个树状数组
套在一起就好啦
暴力开点开不下的
要动态开点

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 100000
inline int read()
{
    int x=0,t=1;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Node
{
    int ls,rs;
    int v;
}t[MAX<<7];
int tot,S[MAX],rt[MAX],n,m,a[MAX];
int t1,t2,tmp1[MAX],tmp2[MAX];
int opt[MAX],ll[MAX],rr[MAX],kk[MAX];
int Tot;
inline int lowbit(int x){return x&(-x);}
void Modify(int &now,int l,int r,int pos,int w)
{
    if(!now)now=++Tot;
    t[now].v+=w;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(pos<=mid)Modify(t[now].ls,l,mid,pos,w);
    else Modify(t[now].rs,mid+1,r,pos,w);
}
void PreModify(int pos,int w){for(int i=pos;i<=n;i+=lowbit(i))Modify(rt[i],1,tot,a[pos],w);}
int Query(int now,int l,int r,int pos)
{
    if(l==r)return t[now].v;
    int mid=(l+r)>>1;
    if(pos<=mid)return Query(t[now].ls,l,mid,pos);
    else return t[t[now].ls].v+Query(t[now].rs,mid+1,r,pos);
}
int Rank(int l,int r,int k)
{
    if(l>r)return 0;
    l-=1;int ret=0;
    for(int i=r;i;i-=lowbit(i))ret+=Query(rt[i],1,tot,k);
    for(int i=l;i;i-=lowbit(i))ret-=Query(rt[i],1,tot,k);
    return ret;
}
int Kth(int l,int r,int k)
{
    if(l==r)return l;
    int sum=0,mid=(l+r)>>1;
    for(int i=1;i<=t1;++i)sum+=t[t[tmp1[i]].ls].v;
    for(int i=1;i<=t2;++i)sum-=t[t[tmp2[i]].ls].v;
    if(sum>=k)
    {
        for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].ls;
        for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].ls;
        return Kth(l,mid,k);
    }
    else
    {
        for(int i=1;i<=t1;++i)tmp1[i]=t[tmp1[i]].rs;
        for(int i=1;i<=t2;++i)tmp2[i]=t[tmp2[i]].rs;
        return Kth(mid+1,r,k-sum);
    }
}
int PreKth(int l,int r,int k)
{
    l--;t1=t2=0;
    for(int i=r;i;i-=lowbit(i))tmp1[++t1]=rt[i];
    for(int i=l;i;i-=lowbit(i))tmp2[++t2]=rt[i];
    return S[Kth(1,tot,k)];
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=n;++i)a[i]=S[++tot]=read();
    for(int i=1;i<=m;++i)
    {
        opt[i]=read();
        if(opt[i]!=3)
        {
            ll[i]=read();rr[i]=read();kk[i]=read();
            if(opt[i]!=2)S[++tot]=kk[i];
        }
        else
        {
            ll[i]=rr[i]=read();
            S[++tot]=kk[i]=read();
        }
    }
    sort(&S[1],&S[tot+1]);
    tot=unique(&S[1],&S[tot+1])-S-1;
    for(int i=1;i<=n;++i)a[i]=lower_bound(&S[1],&S[tot+1],a[i])-S,PreModify(i,1);
    for(int i=1;i<=m;++i)
    {
        if(opt[i]==1)
        {
            kk[i]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
            printf("%d\n",Rank(ll[i],rr[i],kk[i]-1)+1);
        }
        else if(opt[i]==2)
        {
            printf("%d\n",PreKth(ll[i],rr[i],kk[i]));
        }
        else if(opt[i]==3)
        {
            PreModify(ll[i],-1);
            a[ll[i]]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
            PreModify(ll[i],1);
        }
        else if(opt[i]==4)
        {
            kk[i]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
            int gg=Rank(ll[i],rr[i],kk[i]-1);
            if(!gg)puts("-2147483647");
            else printf("%d\n",PreKth(ll[i],rr[i],gg));
        }
        else
        {
            kk[i]=lower_bound(&S[1],&S[tot+1],kk[i])-S;
            int gg=Rank(ll[i],rr[i],kk[i]);
            if(gg==rr[i]-ll[i]+1)puts("2147483647");
            else printf("%d\n",PreKth(ll[i],rr[i],gg+1));
        }
    }
    return 0;
}