luogu P2596 [ZJOI2006]书架

感觉要死在\(Splay\)里了 orz

这题用\(Splay\)维护这个序列,其中的第\(k\)大点代表这个序列的第\(k\)个数

第一个操作,先把那个数所在的点旋到根,然后把整个根的左子树接到右子树中最小的点,记得\(splay\)维护整棵树

第二个操作类似,把第一个操作反过来就行

第三个本质是两个相邻位置交换值,直接把这个数所在的点和前驱/后继交换值

第四个,把对应的点旋到根,输出左子树大小

第五个,就是整棵树的第\(x\)大点上的值

#include<bits/stdc++.h>

#define LL long long

#define il inline

#define re register

using namespace std;

const int N=80000+10;

il int rd()

{

  int x=0,w=1;char ch=0;

  while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}

  while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}

  return x*w;

}

int n,q,b[N],p[N];

int fa[N],ch[N][2],sz[N],a[N],rt,tt;

il void psup(int x){sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1;}

il void rot(int x)

{

  int y=fa[x],z=fa[y],yy=ch[y][1]==x,w=ch[x][!yy];

  ch[z][ch[z][1]==y]=x,fa[x]=z;

  ch[y][yy]=w,fa[w]=y;

  ch[x][!yy]=y,fa[y]=x;

  psup(y),psup(x);

}

il void spl(int x,int en)

{

  while(fa[x]!=en)

    {

      int y=fa[x],z=fa[y];

      if(z!=en) ((ch[y][1]==x)^(ch[z][1]==y))?rot(x):rot(y);

      rot(x);

    }

  if(!en) rt=x;

}

il void inst(int x)

{

  int nw=rt;

  while(ch[nw][x>a[nw]]) nw=ch[nw][x>a[nw]];

  /*

    ++tt,fa[tt]=nw,a[tt]=x;

    if(nw) ch[nw][x>a[nw]]=tt;

    spl(tt,0);

   */

}

il int gkth(int x)

{

  int nw=rt;

  while(233)

    {

      if(x<=sz[ch[nw][0]]) nw=ch[nw][0];

      else if(x>sz[ch[nw][0]]+1) x-=sz[ch[nw][0]]+1,nw=ch[nw][1];

      else return nw;

    }

}

il int bui(int l,int r)

{

  if(l>r) return 0;

  int mid=(l+r)>>1,nw=++tt;

  a[nw]=b[mid],p[b[mid]]=nw,sz[nw]=1;

  if(l==r) return nw;

  fa[ch[nw][0]=bui(l,mid-1)]=nw;

  fa[ch[nw][1]=bui(mid+1,r)]=nw;

  psup(nw);

  return nw;

}

int main()

{

  n=rd(),q=rd();

  for(int i=1;i<=n;i++) b[i]=rd();

  rt=bui(1,n);

  char cc[10];

  while(q--)

    {

      scanf("%s",cc);

      if(cc[0]=='T')

        {

          int x=p[rd()];

          spl(x,0);

          if(!ch[x][0]);

          else if(!ch[x][1]) swap(ch[x][0],ch[x][1]);

          else

            {

              int y=gkth(sz[ch[x][0]]+2);   //后继

              ch[y][0]=ch[x][0],fa[ch[x][0]]=y,ch[x][0]=0;

              spl(y,0);

            }

        }

      else if(cc[0]=='B')

        {

          int x=p[rd()];

          spl(x,0);

          if(!ch[x][1]);

          else if(!ch[x][0]) swap(ch[x][0],ch[x][1]);

          else

            {

              int y=gkth(sz[ch[x][0]]); //前驱

              ch[y][1]=ch[x][1],fa[ch[x][1]]=y,ch[x][1]=0;

              spl(y,0);

            }

        }

      else if(cc[0]=='I')

        {

          int x=p[rd()],c=rd();

          if(c==1)

            {

              spl(x,0);

              int nt=gkth(sz[ch[x][0]]+2);  //主要是这个地方,我本来用普通平衡树里面的求前驱后继,但是这棵树并不是一个权值为关键字的splay,,,

              swap(p[a[x]],p[a[nt]]),swap(a[x],a[nt]);

            }

          else if(c==-1)

            {

              spl(x,0);

              int ft=gkth(sz[ch[x][0]]);

              swap(p[a[x]],p[a[ft]]),swap(a[x],a[ft]);

            }

        }

      else if(cc[0]=='A')

        {

          int x=p[rd()];

          spl(x,0);

          printf("%d\n",sz[ch[x][0]]);

        }

      else printf("%d\n",a[gkth(rd())]);

    }

  return 0;

}

巴特西

luogu P2596 [ZJOI2006]书架

最新文章

热门文章