平衡树学习小结

最近系统的学习了学习平衡树（splay），现做如下总结。

何为平衡树

简而言之，平衡树是一种二叉树。
何为平衡呢？平衡树的“平衡”是由于他的深度较为平衡，一般稳定在log级别。

平衡树的优势

平衡树因其“平衡”而保证了时间复杂度，减少了被卡的概率。

平衡树的基本操作

1.旋转(rotate(x)):将x旋转到他的父亲的位置。

我们如何操作呢？
利用一个性质：平衡树因为中序遍历是有序的，所以他的左子树的所有元素都比他小，右子树都比他大。
设旋转的点为x，他的父亲为y，父亲的父亲为z，son(a)表示a是他父亲的左儿子还是右儿子。
此处分情况讨论。y的左右儿子关系决定了x和z的大小关系。
因为我们要保证平衡树的平衡，所以若y是z的右儿子，那么x必定比z大，所以我们把x接在z的右儿子处，若y是z的左儿子反之。
继续讨论y与x的大小关系：显而易见与“x与y的大小关系”相反，把y接在x后即可。

void rotate(int x)
{
    int y=fa[x],z=fa[y],b=son(x),c=son(y);
    ch[z][c]=x,fa[x]=z;
    if (ch[x][!b]) fa[ch[x][!b]]=y;
    ch[y][b]=ch[x][!b],ch[x][!b]=y,fa[y]=x;
} 

2.Splay(splay(x,p)):将x旋转到p的儿子的位置。

这个操作是依靠rotate实现的。
分三种情况讨论：
1.x和p是爷爷与儿子的关系，即只旋转一次就可以到达位置：rotate(x)
2.son(x)和son(fa[x])不同：两次rotate(x)
3.son(x)和son(fa[x])相同：rotate(fa[x]),rotate(x)
画个图就可以知道为什么。

void splay(int x,int p)
{
    while (fa[x]!=p)
    {
        int y=fa[x],z=fa[y];
        if (z==p)
            rotate(x);
        else
            if (son(x)!=son(y))
                rotate(x),rotate(x);
            else
                rotate(y),rotate(x);
    }
    if (!fa[x]) root=x;
}

3.查找前驱(find_qq(x))、后继(find_hj(x)):查找一个最大的小于（小于等于）x的数、最小的大于（大于等于）x的数。

利用二叉查找树的性质中序遍历有序，我们每次比较当前节点和待查询值的大小关系，决定下一步是往左子树走还是往右子树走。

int find_qq(int x)
{
    ll now=root,ans=0;
    while (now)
        if (val[now]<=x)
            ans=now,now=ch[now][1];
        else
            now=ch[now][0];
    return ans;
}
int find_hj(int x)
{
    ll now=root,ans=0;
    while (now)
        if (val[now]>x)
            ans=now,now=ch[now][0];
        else
            now=ch[now][1];
    return ans;
} 

4.插入(insert(x)):插入一个数到平衡树中。

找到x的前驱和后继，将前驱旋转到根（splay(qq,0)），后继旋转到根的右儿子（splay(hj,qq)），可以看出后继的左儿子一定是空的，因为后继的左儿子小于他并且大于前驱很明显是不存在的。故将新元素插入到后继的左儿子处。

void insert(int x)
{
    int qq=find_qq(x);
    int hj=find_hj(x);
    splay(qq,0);
    splay(hj,qq);
    ch[hj][0]=++cnt;
    fa[cnt]=hj;
    val[cnt]=x;
}

5.删除(del(x)):删除平衡树中已有的元素x。

找到x的前驱，将其旋转到根的位置，再找到x的后继，旋转到根的右儿子的位置，则根的右儿子的左儿子为需删除的元素，清空根的右儿子的左指针即可。

void del(int x)
{
    int qq=find_qq(x-1);
    int hj=find_hj(x);
    splay(qq,0);
    splay(hj,qq);
    ch[hj][0]=0;
}

平衡树的应用

查询元素的前驱后继，查询元素排名，查询排名为k的元素，插入/删除元素等。