模板——伸展树 splay 实现快速分裂合并的序列

Posted 2020-08-23 kiraa

伸展操作：将treap中特定的结点旋转到根

//将序列中从左数第k个元素伸展到根，注意结点键值保存的是原序列id
void splay(Node* &o, int k) {
    int s = o->ch[0] == NULL ? 0 : o->ch[0]->s;
    int d = k <= s ? 0 : (k == s+1 ? -1 : 1);
    if(d == 1) k -= s+1;
    if(d != -1) {
        splay(o->ch[d], k);
        rotate(o, d^1);
    }
}

//合并操作：把序列s1, s2连接在一起，s1在左边
Node* merge(Node* left, Node* right) {
    splay(left, left->s);
    left->ch[1] = right;
    left->maintain();
    return left;
}

//分裂操作：把序列S分成两个连续的子序列，左子序列包含k个元素，右子序列包含剩下的元素
void split(Node* o, int k, Node* &left, Node* &right) {
    splay(o, k);
    left = o;
    right = o->ch[1];
    o->ch[1] = NULL;
    left->maintain();
}