【模板】Splay

【模板】Splay

---

$Splay$，名为伸展树，通过不断将某个节点旋转到根节点，使得整棵树保持$BST$性质的同时，尽量平横。

维护的信息

$rt$	$tot$	$fa[i]$	$ch[i][0/1]$	$val[i]$	$cnt[i]$	$siz[i]$
根结点编号	节点总数	父亲节点	左右儿子节点	节点权值	权值出现次数	子树大小

类定义

class Splay {
   
private:
    int rt, tot, fa[maxn], ch[maxn][2], cnt[maxn], siz[maxn];
public:
    int val[maxn];
};

操作

基本操作

$maintain(x)$ : 更新节点 $x$ 的 $siz$ 值

$get(x)$ : 判断节点 $x$ 是其父亲的左儿子还是右儿子

$clear(x)$ : 清除节点 $x$ 的数据

void maintain(int x) {
    siz[x] = siz[ch[x][0]] + siz[ch[x][1]] + cnt[x]; }
bool get(int x) {
    return x == ch[fa[x]][1]; }
void clear(int x) {
    ch[x][0] = ch[x][1] = fa[x] = cnt[x] = siz[x] = val[x] = 0; }

旋转操作

$rotate(x)$ : 将 $x$ 向上旋转

例如上面这张图，将左图的2右旋，就可以得到右图

旋转的一般操作是（设需要旋转的节点为 $x$ ，其父亲节点为 $y$ ，下面以右旋为例）：

1. 将 $y$ 的左儿子指向 $x$ 的右儿子，$x$ 的右儿子（如果有）的父亲指向 $y$
2. 将 $x$ 的右儿子指向 $y$ ，且 $y$ 的父亲指向 $x$
3. 如果 $y$ 存在父亲 $z$ ，则另原来 $y$ 位置指向 $x$ ，$x$ 的父亲指向 $z$
4. 最后维护节点的 $siz$ ，注意顺序，先 $x$ 后 $y$

void rotate(int x) {
   
    int y = fa[x], z = fa[y], chk = get(x);
    ch[y][chk] = ch[x][chk ^ 1];
    if(ch[x][chk ^ 1]) fa[ch[x][chk ^ 1]] = y;
    ch[x][chk ^ 1] = y;
    fa[y] = x; fa[x] = z;
    if(z) ch[z][y == ch[z][1]] = x;
    maintain(x); maintain(y);
}

伸展操作

$splay(x)$ : 将 $x$ 旋转到根

根据 $splay$ 的性质，每次访问完当前节点后，要强制将其旋转到根节点，情况共有6种：

• $x$ 的父亲为根节点，直接将 $x$ 左旋或右旋

• $x$ 的父亲节点不是根节点，但 $x$ 和父亲的儿子类型相同，先将其父亲左旋或右旋，然后将 $x$ 右旋或左旋

• $x$ 的父亲节点不是根节点，且 $x$ 和父亲的儿子类型不同，将 $x$ 左旋再右旋，或者右旋再左旋

void splay(int x) {
   
    for (int f = fa[x]; f = fa[x], f; rotate(x))
        if(fa[f]) rotate(get(x) == get(f) ? f : x);
    rt = x;
}

插入操作

$insert(x)$ : 将 $x$ 插入到 $splay$ 中

插入分以下几种情况：

• 树为空，直接插入根节点
• 当前节点的 $val$ 等于 $k$ ，增加当前节点的大小，并更新其和其父亲节点的信息，对当前节点 $splay$
• 按照 $BST$ 性质向下找，找到空节点插入即可

void insert(int k) {
   
    if(!rt) {
   
        val[++tot] = k;
        cnt[tot]++;
        rt = tot;
        maintain(rt);
        return;
    }
    int cur = rt, f = 0