[BZOJ3595][SCOI2014]方伯伯的OJ(平衡树)

题目:

我是超链接

题解:

单看操作来说都是平衡树基本操作,然后范围呢?1e8。。但是操作数只有1e5,我们就要考虑从操作数下手

思路是把编号和排名连续的点并到平衡树的一个节点中维护,有一个操作就把一个完整节点分裂,这样复杂度就只与修改次数有关了。

排名和编号的维护是这样的。

平衡树节点按排名维护,由于维护区间,键值要记l[x]和r[x]两个。初始时只有一个节点l=1,r=n。又由于调换操作只有调到首尾,所以节点键值可以离散,记当前键值的min和max,放到前面就是–min,放到后面就是++max。

每个节点再记一下对应的编号,而且如果节点维护的是一段区间,说明中间还没被修改过,其编号就是键值。序列中的节点再用一个map记录一下对应平衡树中的键值,只有修改了才记,没有就是编号本身。

代码:

#include <map>
#include <cstdio>
using namespace std;
const int N=300005;
int f[N],size[N],key[N],pos[N],ch[N][2],L[N],R[N];
int root,sz,n,m;map<int,int> rak;
int get(int x){return ch[f[x]][1]==x;}
void updata(int now){size[now]=size[ch[now][0]]+size[ch[now][1]]+R[now]-L[now]+1;}
void rotate(int x)
{
    int old=f[x],oldf=f[old],which=get(x);
    f[x]=oldf; if (oldf) ch[oldf][get(old)]=x;
    ch[old][which]=ch[x][which^1]; f[ch[x][which^1]]=old;
    ch[x][which^1]=old; f[old]=x;
    updata(old); updata(x);
}
void splay(int x,int to)
{
    for (;f[x]!=to;rotate(x))
      if (f[f[x]]!=to) rotate(get(x)==get(f[x])?f[x]:x);
    if(!to) root=x;
}
void insert(int &now,int fa,int l,int r,int v)
{
    if (l>r) return;
    int mid=(l+r)>>1;
    if (!now)
    {
        now=++sz; key[now]=v; f[now]=fa;
        size[now]=r-l+1; L[now]=l; R[now]=r;
        splay(now,0);
        return;
    }
    if (r<L[now]) insert(ch[now][0],now,l,r,v);
    else insert(ch[now][1],now,l,r,v);
    if (now) updata(now);
}
int nxt()
{
    int now=ch[root][1];
    while (ch[now][0]) now=ch[now][0];
    return now;
}
int find(int now,int v)
{
    if (L[now]<=v && v<=R[now]) return now;
    if (L[now]>v) return find(ch[now][0],v);
    else return find(ch[now][1],v);
}
int find_id(int now,int x)
{
    if (x>size[ch[now][0]] && x<=size[now]-size[ch[now][1]])
    {
        if (L[now]==R[now]) return key[now];
        return x-size[ch[now][0]]+L[now]-1;
    }
    if (x<=size[ch[now][0]]) return find_id(ch[now][0],x);
    else return find_id(ch[now][1],x-(size[now]-size[ch[now][1]]));
}
void del(int now,int ra)
{
    if (!ch[now][1]) root=ch[now][0],f[root]=0;
    else
    {
        int y=nxt(); splay(y,now);
        root=y; f[y]=0;
        if (ch[now][0]) ch[y][0]=ch[now][0],f[ch[now][0]]=y;
        updata(y);
    }
    insert(root,0,L[now],ra-1,L[now]); insert(root,0,ra+1,R[now],ra+1); 
}
void split(int now,int ra,int id)//now这个节点,排第ra的权值修改为id 
{
    if (L[now]==R[now]) key[now]=id;
    else del(now,ra),insert(root,0,ra,ra,id);
}
int main()
{
    scanf("%d%d",&n,&m); 
    int minn=0,maxx=n;
    insert(root,0,1,n,1);int ans=0;
    for (int i=1;i<=m;i++)
    {
        int id,x,y;
        scanf("%d%d",&id,&x);x-=ans;
        if (id==1)
        {
            scanf("%d",&y);y-=ans;
            int r=rak[x]; if (!r) r=x;//编号为x的排名 
            int now=find(root,r); splay(now,0);
            ans=size[ch[root][0]]+r-L[now]+1;
            printf("%d\n",ans);

            split(now,r,y); rak[x]=0; rak[y]=r;
        }else if(id<4) 
        {
            int r=rak[x]; if (!r) r=x;
            int now=find(root,r); splay(now,0);
            ans=size[ch[root][0]]+r-L[now]+1;
            printf("%d\n",ans);
            //将编号转化为now节点

            del(now,r); r=rak[x]=((id==2)?--minn:++maxx);
            insert(root,0,r,r,x);
        }
        else ans=find_id(root,x),printf("%d\n",ans);
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值