splay小结—植树结

我要把高级数据结构当爸爸了... ...弱到跪烂了。

splay,二叉搜索树的一种,具有稳定变形功能

  二叉搜索树:对于一个节点,都只有不超过2个孩子。其左子树内的点的权值都比这个点小,右子树的点的权值都比这个点要大(等于的话随你)。这个性质对于所有点都成立。

  我们可以看到二叉搜索树适用于解决求前驱后继、求排名、求第k大等问题。但是如果出题人执(sang)意(xin)要(bing)卡(kuang),造出递减或递增的数据出来,那么你的常规二叉搜索树的每一次插入和查找都是O(n)的,会被时代淘汰的。我们只有提高自己的姿势水平才行。

  splay也是一颗二叉搜索树。但是它有个优点:它可以通过旋转节点改变树的形状,江低树的直径,从而江低复杂度。从理论上看,这棵树的深度是logN的,所以一次查询的复杂度就是logN。

  那么怎样进行旋转节点就是一个神奇的问题了!

  我们要把节点x旋转到它的父亲y的位置,就可以找到规律:

    如果x是左儿子,那么y就要变成x的右儿子,x原本的右儿子要变成y的左儿子。

    如果x是右儿子,情况是差不多的。

  这点东西都很好写,几个if就差不多了(所以常数很大)。关键是旋转技巧。

  splay(x,goal),表示江x旋转到goal的儿子处。

    如果x是goal的孙子,那么转一次就好了。

    不是的话,我们就可以把x转到它的爷爷的位置上面(因为要江低复杂度和树的直径,别问我为什么,问塔尖去)。

  你还可以在旋转顺序上做点优化来达到江低直径的效果。

  上题。BZOJ书架 操作还是很全的。

  但其实初学者还是像我一样先做 codevs营业额统计 比较好,操作较少但splay不少。

  改变位置就相当于删除再插入。求排名就是把它splay到顶端然后看左子树大小。

  第k本就是第k大,就是二叉搜索树的普通操作。

  书架 代码如下。num应该改名成fact,表示点对应的书的编号。

#include    <iostream>
#include    <cstdio>
#include    <cstdlib>
#include    <algorithm>
#include    <vector>
#include    <cstring>
#include    <queue>
#define LL long long int
#define ls (x << 1)
#define rs (x << 1 | 1)
using namespace std;
const int N = 300010;
int n,m,T[N],ch[N][2],size[N],num[N];
int fa[N],root,tot,pos[N],rec[N];
int gi()
{
  int x=0,res=1;char ch=getchar();
  while(ch>'9'||ch<'0'){if(ch=='-')res*=-1;ch=getchar();}
  while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
  return x*res;
}
LL gl()
{
  LL x=0,res=1;char ch=getchar();
  while(ch>'9'||ch<'0'){if(ch=='-')res*=-1;ch=getchar();}
  while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
  return x*res;
}
int gsize(int x){return x?size[x]:x;}
void up(int x){size[x]=gsize(ch[x][0])+gsize(ch[x][1])+1;}
void rtt(int x,int kind)  //综合版,包含了zig和zag。
{
  int y=fa[x],z=fa[y];
  ch[y][!kind]=ch[x][kind];fa[x]=z;
  if(ch[x][kind])fa[ch[x][kind]]=y;
  if(z)
    if(ch[z][0]==y)ch[z][0]=x;
    else ch[z][1]=x;
  ch[x][kind]=y;fa[y]=x;
  up(y);up(x);
}
void splay(int x,int goal)
{
  if(x==goal)return;
  while(fa[x]!=goal)
    if(fa[fa[x]]==goal)
      rtt(x,ch[fa[x]][0]==x);
    else
      {
	int y=fa[x],kind=ch[fa[y]][0]==y;
	if(ch[y][kind]==x)rtt(x,kind^1),rtt(x,kind);
	else rtt(y,kind),rtt(x,kind);  //听缩这样可以江低复杂度,实际上是的。
      }
  if(goal==0)root=x;
}
void newnode(int &r,int fat,int x)
{
  r=++tot;fa[tot]=fat;size[tot]=1;
  ch[tot][0]=ch[tot][1]=0;num[tot]=x;return;
}
void insert(int rt,int x,int value)   //把一个编号为x的书放到位置value上
{
  if(value==1 && ch[rt][0]==0){newnode(ch[rt][0],rt,x);return;}
  if(value==2+gsize(ch[rt][0])&&ch[rt][1]==0){newnode(ch[rt][1],rt,x);return;}
  if(value<=1+gsize(ch[rt][0]))insert(ch[rt][0],x,value);
  else insert(ch[rt][1],x,value-gsize(ch[rt][0])-1);
  up(x);
}
void del(int x)
{
  splay(x,0);
  if(ch[x][0]==0&&ch[x][1]==0){root=0;return;}
  if(ch[x][0]==0){root=ch[x][1];fa[root]=0;return;}
  if(ch[x][1]==0){root=ch[x][0];fa[root]=0;return;}
  int tmp=ch[x][0];while(ch[tmp][1])tmp=ch[tmp][1];
  splay(tmp,x);root=tmp;fa[tmp]=0;
  ch[tmp][1]=ch[x][1];fa[ch[x][1]]=tmp;
  up(tmp);
}
int getsize(int x){splay(x,0);return gsize(ch[x][0])+1;}
int gc()
{
  char ch=getchar();
  while(ch>'Z' || ch<'A')ch=getchar();
  if(ch=='T')return 1;
  if(ch=='B')return 2;
  if(ch=='I')return 3;
  if(ch=='A')return 4;
  if(ch=='Q')return 5;
  return 0;
}
void pr(int x)
{
  printf(" point %d:  ls=%d  rs=%d  size=%d
",x,ch[x][0],ch[x][1],gsize(x));
  if(ch[x][0])pr(ch[x][0]);if(ch[x][1])pr(ch[x][1]);
}
int find(int x)
{
  int pt=x,r=root;
  while(1)
    {
      if(pt==1+gsize(ch[r][0]))return num[r];
      if(pt>1+gsize(ch[r][0]))pt-=(1+gsize(ch[r][0])),r=ch[r][1];
      else r=ch[r][0];
    }
  return num[0];
}
int main()
{
  n=gi();m=gi();
  for(int i=1;i<=n;++i)
    {
      rec[i]=gi();
      pos[rec[i]]=i;
      if(i==1)newnode(root,0,rec[1]),root=tot;
      else{insert(root,rec[i],i);if(i%2)splay(tot,0);}
    }
  for(int i=1;i<=m;++i)
    {
      int type=gc(),x=gi(),y=type==3?gi():0;
      switch(type)
	{
	case 1:{del(pos[x]);insert(root,x,1);pos[x]=tot;splay(tot,0);break;}
	case 2:{del(pos[x]);insert(root,x,n);pos[x]=tot;splay(tot,0);break;}
	case 3:{int S=getsize(pos[x])+y;del(pos[x]);insert(root,x,S);splay(tot,0);pos[x]=tot;break;}
	case 4:{printf("%d
",getsize(pos[x])-1);break;}
	case 5:{printf("%d
",find(x));break;}
	default:continue;
	}
    }
  return 0;
}

  

原文地址:https://www.cnblogs.com/fenghaoran/p/6603449.html