[平衡树模板]Treap

算法标签 Treap

种下第一棵平衡树…

题目描述 Description

这是一道模板题。

如果觉得这个题水的可以做一下4544压行，是千古神犇花爸爸出的神犇题。

您需要写一种数据结构（可参考题目标题，但是这句话其实并没有什么用233），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

输入描述 Input Description

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)

输出描述 Output Description

对于操作3,4,5,6每行输出一个数，表示对应答案

注意：这道题重复数据可以加入Treap

#include <cstdio>#include <cstdlib>#include <algorithm>#include <ctime>using namespace std;const int maxn = 100005;struct node {    node *ch[2];    int r, v, s; //v-键值 r-优先级 s - 子树结点个数    inline int cmp(int x) const {      return x < v ? 0 : 1;    }    inline void maintain() { //计算子结点个数      s = 1;      if(ch[0] != NULL) s += ch[0] -> s;      if(ch[1] != NULL) s += ch[1] -> s;    }};void retate(node* &x, int d) //旋转, d = 0 左旋 d = 1 右旋{    node *k = x -> ch[d ^ 1];    x->ch[d ^ 1] = k -> ch[d]; k->ch[d] = x;    x->maintain(); k->maintain(); //重新计算子结点个数    x = k;}void insert(node* &x, int k) //在以x为根的子树中插入键值k，修改x{    if(x == NULL) {x = new node(); x -> ch[0] = x -> ch[1] = NULL; x -> v = k; x -> r = rand(); x -> s = 1; }//x为空    else {      int d = x -> cmp(k);      insert(x -> ch[d], k); if(x -> ch[d] -> r > x -> r) retate(x, d ^ 1);     }    x -> maintain();}void remove(node* &x, int k) //在以x为根的子树中删除键值k，修改x{    if(x -> v == k) { //k = x.v      if(x -> ch[0] == NULL) x = x -> ch[1]; //左子树为空      else if(x -> ch[1] == NULL) x = x -> ch[0]; //右子树为空      else {        int d = (x -> ch[0] -> r > x -> ch[1] -> r ? 1 : 0);        retate(x, d); remove(x -> ch[d], k); //递归删除      }    }else remove(x -> ch[x -> cmp(k)], k);    if(x != NULL) x -> maintain();}int Rank(node *x, int k) //查找k的排名{    if(x == NULL) return 1;    if(k <= x -> v) return Rank(x ->ch[0], k);    else return Rank(x -> ch[1], k) + (x -> ch[0] != NULL ? x -> ch[0] -> s : 0) + 1; }int kth(node *x, int k) //查找第k小元素{    if(x == NULL || k <= 0 || k > x -> s) return 0;    int s = (x -> ch[0] == NULL ? 0 : x -> ch[0] -> s);    if(s + 1 == k) return x -> v;    else if(k <= s) return kth(x -> ch[0], k);    else return kth(x -> ch[1], k - s - 1);}int last(node *x, int k) //求k的前驱{    if(x == NULL) return -(1 << 30);    if(x -> v < k) return max(x -> v, last(x -> ch[1], k));    else return last(x -> ch[0], k);}int next(node *x, int k) //求k的后继{    if(x == NULL) return (1 << 30);    if(x -> v > k) return min(x -> v, next(x -> ch[0], k));    else return next(x -> ch[1], k);}int main(){    freopen("phs.in", "r", stdin);    freopen("phs.out", "w", stdout);    int cnt = 0, n; scanf("%d", &n);    node *root = NULL;    for(int x, opt, i = 1; i <= n; ++i) {      scanf("%d%d", &opt, &x);      if(opt == 1) insert(root, x);      else if(opt == 2) remove(root, x);      else if(opt == 3) printf("%d\n", Rank(root, x));      else if(opt == 4) printf("%d\n", kth(root, x));      else if(opt == 5) printf("%d\n", last(root, x));      else if(opt == 6) printf("%d\n", next(root, x));    }    return 0;}