【BZOJ 3196】 Tyvj 1730 二逼平衡树

来源:互联网 发布:跳转页面js 编辑:程序博客网 时间:2024/05/17 23:09

终于填坑!!!!!!
鬼知道以前怎么错的!!
反正现在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;}
0 0