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

最新推荐文章于 2020-07-31 10:09:42 发布

转载最新推荐文章于 2020-07-31 10:09:42 发布 · 204 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Kiraa/p/6192507.html

本文介绍Treap数据结构中的伸展与合并操作实现。伸展操作通过旋转将特定节点移动到根位置，合并操作则连接两个序列，确保左序列位于右侧序列之前。文章还提供了分裂操作的实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

伸展操作：将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();
}