bzoj 3196 树套树模板

    然而我还是在继续刷水题。。。

    终于解开了区间第k大的心结。。。

    比较裸的线段树套平衡树，比较不好想的是求区间第k大时需要二分一下答案，然后问题就转化为了第一个操作。复杂度nlog3n。跑的比较慢。。。

    在查前驱后继的时候写错了。。。如果要直接赋值ans的话前驱是k[x]<=z,后继是k[x]<z,如果都写<的话需要取max和min。。。（不是第一次犯这种错了）

   

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lc x*2,l,mid
#define rc x*2+1,mid+1,r
#define ls(x) ch[x][0]
#define rs(x) ch[x][1]
#define inf 0x3f3f3f3f
using namespace std;
int n,m;
int root[50005*50];
int size[50005*50],k[50005*50];
int ch[50005*50][2],fa[50005*50];
int a[50005];
int cnt;
inline void push_up(int x)
{
    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
    return ;
}
inline void rotate(int p)
{
    int q=fa[p],y=fa[q],x=(ch[q][1]==p);
    ch[q][x]=ch[p][x^1];fa[ch[q][x]]=q;
    ch[p][x^1]=q;fa[q]=p;fa[p]=y;
    if(y)
    {
        if(ls(y)==q)ch[y][0]=p;
        else ch[y][1]=p;
    }
    push_up(q);push_up(p);
}
inline void splay(int now,int x)
{
    for(int y;y=fa[x];rotate(x))
    {
        if(fa[y])
        {
            if((ls(y)==x&&ls(fa[y])==y)||(rs(y)==x&&rs(fa[y])==y))rotate(y);
            else rotate(x);
        }
    }
    root[now]=x;
}
inline void insert(int now,int x,int z)
{
    size[x]++;
    while(ch[x][k[x]<z])x=ch[x][k[x]<z],size[x]++;
    ch[x][k[x]<z]=++cnt;fa[cnt]=x;size[cnt]=1;k[cnt]=z;splay(now,cnt);
}
inline void build(int x,int l,int r)
{
    root[x]=++cnt;k[cnt]=a[l];size[cnt]=1;
    for(int i=l+1;i<=r;i++)insert(x,root[x],a[i]);
    int mid=(l+r)>>1;
    if(l<r)build(lc),build(rc);
    return ; 
}
inline int fd(int x,int z)
{
   int ans=0;
   while(x)    
   {
         if(k[x]>z)
         {
              x=ch[x][0];
      }
      else ans+=size[ch[x][0]]+1,x=ch[x][1];
   }
   return ans;
}
inline int qur(int x,int l,int r,int ll,int rr,int kk)
{
    if(ll<=l&&rr>=r)
    {
        return fd(root[x],kk);
    }
    int mid=(l+r)>>1;
    int ans=0;
    if(ll<=mid)ans+=qur(lc,ll,rr,kk);
    if(rr>mid)ans+=qur(rc,ll,rr,kk);
    return ans;
}
inline int find(int x,int z)
{
    if(k[x]==z)return x;
    while(ch[x][k[x]<z])
    {
        x=ch[x][k[x]<z];if(k[x]==z)return x;
    }
    return 0;
}
inline void del(int now,int x)
{
    splay(now,x);
    if(!ch[x][0])root[now]=ch[x][1],fa[ch[x][1]]=0;
    else if(!ch[x][1])root[now]=ch[x][0],fa[ch[x][0]]=0;
    else 
    {
        fa[ch[x][0]]=0;int tmp=ch[x][0];
        while(ch[tmp][1])tmp=ch[tmp][1];
        splay(now,tmp);ch[tmp][1]=ch[x][1];fa[ch[x][1]]=tmp;push_up(tmp);
    }
}
inline void gai(int x,int l,int r,int pos,int z)
{
    insert(x,root[x],z);
    int x1=find(root[x],a[pos]);
    del(x,x1);
    if(l==r)return ;
    int mid=(l+r)>>1;
    if(pos<=mid)gai(lc,pos,z);
    else gai(rc,pos,z);
}
inline int pre(int x,int z)
{
    int ans=-inf;
    while(ch[x][k[x]<=z])
    {
        if(k[x]<=z)ans=k[x];
        x=ch[x][k[x]<=z];
    }if(k[x]<=z)ans=max(ans,k[x]);
    return ans;
}
inline int suc(int x,int z)
{
    int ans=inf;
    while(ch[x][k[x]<z])
    {
        if(k[x]>=z)ans=k[x];
        x=ch[x][k[x]<z];
    }if(k[x]>=z)ans=min(ans,k[x]);
    return ans;
}
inline int qur_pre(int x,int l,int r,int ll,int rr,int kk)
{
    if(l>=ll&&r<=rr)
    {
        return pre(root[x],kk);
    }
    int ans=-inf;
    int mid=(l+r)>>1;
    if(ll<=mid)ans=max(ans,qur_pre(lc,ll,rr,kk));
    if(rr>mid)ans=max(ans,qur_pre(rc,ll,rr,kk));
    return ans;
}
inline int qur_suc(int x,int l,int r,int ll,int rr,int kk)
{
    if(l>=ll&&r<=rr)
    {
        return suc(root[x],kk);
    }
    int ans=inf;
    int mid=(l+r)>>1;
    if(ll<=mid)ans=min(ans,qur_suc(lc,ll,rr,kk));
    if(rr>mid)ans=min(ans,qur_suc(rc,ll,rr,kk));
    return ans;
}
int mn,mx;
int main()
{
    scanf("%d%d",&n,&m);mn=inf;mx=-inf;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);mn=min(mn,a[i]);mx=max(mx,a[i]);
    }
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int t1;
        scanf("%d",&t1);
        int l,r,kk;
        if(t1==1)
        { 
           scanf("%d%d%d",&l,&r,&kk);
           printf("%d\n",qur(1,1,n,l,r,kk-1)+1);
        }
        else if(t1==2)
        {
            scanf("%d%d%d",&l,&r,&kk);
            int ha=mn;int ta=mx;
            while(ha<=ta)
            {
                int mid=(ha+ta)>>1;
                if(qur(1,1,n,l,r,mid)<kk)ha=mid+1;
                else ta=mid-1;
            }
            printf("%d\n",ha);
        }
        else if(t1==3)
        {
            scanf("%d%d",&l,&kk);
            gai(1,1,n,l,kk);a[l]=kk;
        }
        else if(t1==4)
        {
            scanf("%d%d%d",&l,&r,&kk);
            printf("%d\n",qur_pre(1,1,n,l,r,kk-1));
        }
        else 
        {
            scanf("%d%d%d",&l,&r,&kk);
            printf("%d\n",qur_suc(1,1,n,l,r,kk+1));
        }
    }
    return 0;
}