【BZOJ 3196】 Tyvj 1730 二逼平衡树

终于填坑!!!!!!
鬼知道以前怎么错的!!
反正现在A了
虽然tyvj还是80 但我相信 那是splay的问题 换做treap就可以(虽然不会....

以后写代码 不要怕 长名变量 取有意义的名字

要以易于维护为第一要求 其次再是美观!!

这次有很多地方可以省略 但为了可维护 就不要怕麻烦 反正比WA掉强!!!

http://pan.baidu.com/s/1nuta3xf 用金币在Tyvj扣的数据 2组大 2组小 我还对着小数据手动排序【捂脸 其实自己通读一遍代码就差不多!

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <algorithm>
using namespace std;

int init[50000+1];
int seg_num[50000*4];
int num[2000000],ch[2000000][2],fa[2000000],size[2000000],root[50000*4],cnt=0;
int n,m,opt;
int ans;
void vist(int now)
{
    if(now==0) return ;
    cout<<num[now]<<' '<<size[now]<<' '<<num[ch[now][0]]<<' '<<num[ch[now][1]]<<endl;
    vist(ch[now][0]);
    vist(ch[now][1]);
}
void Up(int x)
{
    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;
}
void Rotate(int x,int &rt)
{
    int y=fa[x],z=fa[y];
    if(y==rt) 
    {
        rt=x;
        z=0;///
    }
    else
    {
        ch[z][ ch[z][1]==y ]=x;
    }
    int c=ch[y][1]==x;
    fa[x]=z;fa[y]=x;fa[ch[x][c^1]]=y;
    ch[y][c]=ch[x][c^1];ch[x][c^1]=y;
    Up(y);Up(x);
}
void Splay(int x,int &rt)
{
    while(x!=rt)
    {
        if(fa[x]!=rt) Rotate(fa[x],rt);
        Rotate(x,rt);
    }
}
void Ins(int &now,int k,int last,int &rt)
{
    if(now==0)
    {
        now=++cnt;
        fa[now]=last;
        size[now]=1;
        num[now]=k;
        
        Splay(now,rt);
        return ;
    } 
    if(k>=num[now]) Ins(ch[now][1],k,now,rt);
    else Ins(ch[now][0],k,now,rt);
}
void Build(int now,int L,int R)
{
    for(int i=L;i<=R;i++) Ins(root[now],init[i],root[now],root[now]);
    
    if(L==R) return;
    
    Build(now*2,L,(L+R)/2);
    Build(now*2+1,(L+R)/2+1,R);
}
void Sp_Rank(int now,int k)
{

    if(now==0) return ;
    if(num[now]>=k) 
    {
        Sp_Rank(ch[now][0],k);
    }
    else
    {
        ans+=size[ch[now][0]]+1;
        Sp_Rank(ch[now][1],k);
    }
}
void Seg_Rank(int now,int L,int R,int s,int t,int k)
{
    if(s<=L&&R<=t)
    {
        Sp_Rank(root[now],k);
        return ;
    }
    int mid=(L+R)/2;
    if(s<=mid) Seg_Rank(now*2,L,mid,s,t,k);
    if(mid+1<=t) Seg_Rank(now*2+1,mid+1,R,s,t,k);
}
int Kth(int s,int t,int k)
{
    int L=0,R=100000000;
    int ou;
    while(L<R)
    {
        int mid=(L+R)/2;
        ans=1;
        Seg_Rank(1,1,n,s,t,mid);
        if(ans<=k) ou=mid,L=mid+1;
        else R=mid;
    }
    return ou;
}
int Find(int now,int x)
{
    if(num[now]==x) return now;
    if(x<num[now]) return Find(ch[now][0],x);
    else return Find(ch[now][1],x);
}
void Sp_Change(int &rt,int v,int k)
{
    int loc=Find(rt,v);
    Splay(loc,rt);
    if(ch[rt][0]*ch[rt][1]==0) rt=ch[rt][0]+ch[rt][1];
    else
    {
        int tmp=ch[rt][0];
        while(ch[tmp][1]!=0) tmp=ch[tmp][1];
        Splay(tmp,ch[rt][0]); ch[ch[rt][0]][1]=ch[rt][1];
        size[ch[rt][0]]+=size[ch[rt][1]];
        fa[ch[rt][1]]=ch[rt][0];
        rt=ch[rt][0];
    }
    Ins(rt,k,rt,rt); 
}
void Seg_Change(int now,int L,int R,int pos,int k)
{
    Sp_Change(root[now],init[pos],k);
    if(L==R) return ;
    int mid=(L+R)/2;
    if(pos<=mid) Seg_Change(now*2,L,mid,pos,k);
    else Seg_Change(now*2+1,mid+1,R,pos,k); 
}
void Sp_Before(int now,int k)
{
    if(now==0) return ;
    if(num[now]>=k) Sp_Before(ch[now][0],k);
    else ans=max(ans,num[now]),Sp_Before(ch[now][1],k);
}
void Seg_Before(int now,int L,int R,int s,int t,int k)
{
    if(s<=L&&R<=t) 
    {
        Sp_Before(root[now],k);
        return ;
    }
    int mid=(L+R)/2;
    if(s<=mid) Seg_Before(now*2,L,mid,s,t,k);
    if(mid+1<=t) Seg_Before(now*2+1,mid+1,R,s,t,k);
}
void Sp_After(int now,int k)
{
    if(now==0) return ;
    if(num[now]<=k) Sp_After(ch[now][1],k);
    else ans=min(ans,num[now]),Sp_After(ch[now][0],k);
}
void Seg_After(int now,int L,int R,int s,int t,int k)
{
    if(s<=L&&R<=t) 
    {
        Sp_After(root[now],k);
        return ;
    }
    int mid=(L+R)/2;
    if(s<=mid) Seg_After(now*2,L,mid,s,t,k);
    if(mid+1<=t) Seg_After(now*2+1,mid+1,R,s,t,k);
}
int main()
{
//  freopen("a.in","r",stdin);
    
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&init[i]); 
    Build(1,1,n);
    
    int L,R,pos,k;
    while(m--)
    {
        scanf("%d",&opt);
        switch (opt)
        {
            case 1:
                {
                    scanf("%d %d %d",&L,&R,&k);
                    ans=0;
                    Seg_Rank(1,1,n,L,R,k);
                    printf("%d\n",ans+1);
                    break;
                }
            case 2:
                {
                    scanf("%d %d %d",&L,&R,&k);
                    printf("%d\n",Kth(L,R,k));
                    break;
                }
            case 3:
                {
                    scanf("%d %d",&pos,&k);
                    Seg_Change(1,1,n,pos,k);
                    init[pos]=k;
                    break;
                } 
            case 4:
                {
                    scanf("%d %d %d",&L,&R,&k);
                    ans=-1;
                    Seg_Before(1,1,n,L,R,k);
                    printf("%d\n",ans);
                    break;
                }
            case 5:
                {
                    scanf("%d %d %d",&L,&R,&k);
                    ans=100000000+1;
                    Seg_After(1,1,n,L,R,k);
                    printf("%d\n",ans);
                    break;
                }
        }
    }
    
    return 0;
}

你可能感兴趣的:(【BZOJ 3196】 Tyvj 1730 二逼平衡树)