[ZJOI2006]书架

题目描述:

1. Top S——表示把编号为S的书放在最上面。

2. Bottom S——表示把编号为S的书放在最下面。

3. Insert S T——T∈{-1,0,1},若编号为S的书上面有X本书,则这条命令表示把这本书放回去后它的上面有X+T本书;

4. Ask S——询问编号为S的书的上面目前有多少本书。

5. Query S——询问从上面数起的第S本书的编号。

题目分析:

1、 将某元素置顶:将元素旋到根,然后将左子树合并到该元素的后继

2、 将某元素置底:将元素旋到根,然后将右子树合并到该元素的前驱

3、 将某元素提前/滞后1位:直接与该元素的前驱/后继交换位置及信息

4、 询问指定元素排名:将元素旋到根,输出size-1

5、 询问指定排名元素:在树上find

题目链接:

Luogu 2596
BZOJ 1861

Ac 代码:

#include <cstdio>
#include <iostream>
#define il inline
const int maxm=81000;
int pos[maxm],v[maxm],ch[maxm][2],fa[maxm],siz[maxm];
int rt,sz;
int n,m;
il void xlclear(int x)
{
    v[x]=fa[x]=ch[x][0]=ch[x][1]=siz[x]=0;
}
il void pushup(int x)
{
    siz[x]=siz[ch[x][0]]+siz[ch[x][1]]+1;
    pos[v[ch[x][0]]]=ch[x][0],pos[v[ch[x][1]]]=ch[x][1];
}
il int get(int x)
{
    return x==ch[fa[x]][1];
}
il void rotate(int x)
{
    int f1=fa[x],f2=fa[f1];
    int g1=get(x),g2=get(f1);
    int t=ch[x][g1^1];
    ch[f1][g1]=t;
    fa[t]=f1;
    fa[f1]=x;
    ch[x][g1^1]=f1;
    fa[x]=f2;
    if(f2) ch[f2][g2]=x;
    pushup(f1),pushup(x);
}
il void splay(int x,int top)
{
    for(int fax;(fax=fa[x])!=top;rotate(x))
     if(fa[fax]!=top)
      rotate(get(fax)==get(x)?fax:x);
    if(!top) rt=x;
    pos[v[x]]=x;
}
il int find(int x,int id)
{
    if (!x) return 0;
    if(siz[ch[x][0]]+1==id) return x;
    if(siz[ch[x][0]]+1>id) return find(ch[x][0],id);
    else return find(ch[x][1],id-siz[ch[x][0]]-1);
}
il void insert(int x)
{
   v[++sz]=x,siz[sz]=1,pos[x]=sz,ch[sz][0]=ch[sz][1]=0;
   if(!rt) rt=sz;
   if(sz>1)
   {
     ch[sz-1][1]=sz;fa[sz]=sz-1;
     //if(sz==6) printf("%d\n",fa[6]);
     splay(sz,0);
     //if(sz==6) printf("%d\n",fa[6]);
   }
}
il void top(int x)
{
    x=pos[x];
    splay(x,0); 
    //printf("-%d-\n",x);
    if (!ch[x][0]) return;
    if (!ch[x][1]) ch[x][1]=ch[x][0],ch[x][0]=0;
    else
    {
        int y=find(rt,siz[ch[x][0]]+2);
        fa[ch[rt][0]]=y;
        ch[y][0]=ch[rt][0];
        ch[rt][0]=0;
        splay(y,0);
    }
}
il void down(int x)
{
    x=pos[x];
    splay(x,0);
    if (!ch[x][1]) return;
    if (!ch[x][0]) ch[x][0]=ch[x][1],ch[x][1]=0;
    else
    {
        int y=find(rt,siz[ch[x][0]]);
        fa[ch[rt][1]]=y;
        ch[y][1]=ch[rt][1];
        ch[rt][1]=0;
        splay(y,0);
    }
}
void ins(int x,int posx)
{
    if (!posx) return;
    splay(pos[x],0);

    int y=find(rt,posx==1?siz[ch[pos[x]][0]]+2:siz[ch[pos[x]][0]]);
    int x1=v[y],x2=pos[x];
    std::swap(pos[x],pos[x1]);
    std::swap(v[x2],v[y]);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1,x;i<=n;i++) scanf("%d",&x),insert(x);
    for(int i=1;i<=m;i++)
    {
       char s[20];
       scanf("%s",s);
       if(s[0]=='I')
       {
         int x,t;
         scanf("%d%d",&x,&t);
         ins(x,t);
       }
       else
       {
         int x;
         scanf("%d",&x);
         if(s[0]=='T') top(x);
         if(s[0]=='B') down(x);
         if(s[0]=='A')
         {
            splay(pos[x],0);
            printf("%d\n",siz[ch[rt][0]]);
         }
         if(s[0]=='Q')
          printf("%d\n",v[find(rt,x)]);
       }

    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值