summaryrefslogtreecommitdiff
path: root/datastructures/treap.cpp
blob: 7c4ce7fb07b696fb3014dc831ce3d9120571ee62 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
struct item {
  int key, prior;
  item *l, *r;
  item() { }
  item (int key, int prior) : key(key), prior(prior), l(NULL), r(NULL) { }
};

void split (item *t, int key, item *l, item *r) {
  if (!t) l = r = NULL;
  else if (key < t->key) split(t->l, key, l, t->l), r = t;
  else split(t->r, key, t->r, r), l = t;
}

void insert (item *t, item *it) {
  if (!t) t = it;
  else if (it->prior > t->prior) split(t, it->key, it->l, it->r), t = it;
  else insert(it->key < t->key ? t->l : t->r, it);
}

void merge (item *t, item *l, item *r) {
  if (!l || !r) t = l ? l : r;
  else if (l->prior > r->prior) merge(l->r, l->r, r), t = l;
  else merge(r->l, l, r->l), t = r;
}

void erase (item *t, int key) {
  if (t->key == key) merge (t, t->l, t->r);
  else erase(key < t->key ? t->l : t->r, key);
}

item *unite (item *l, item *r) {
  if (!l || !r)  return l ? l : r;
  if (l->prior < r->prior) swap(l, r);
  item * lt, rt;
  split(r, l->key, lt, rt);
  l->l = unite(l->l, lt);
  l->r = unite(l->r, rt);
  return l;
}