问题描述
我们知道,Treap可以完成节点的动态插入、删除、查询,其每个操作的时间复杂度是O(log n),因为其实现较红黑树更为简单,因此常常用于某些场合,以替换红黑树的实现。
Treap的每个节点维护了key, priority。
struct Node {
int key;
int priority;
Node (int k, int p): key(k), priority(p) {}
}
key是作为BST的键值,用于支持快速的插入、删除和查询操作,而priority则是用于维护最小堆的性质。即从key层面上看,整个树是一个BST;从priority层面上看,则整个树是一个最小堆。
关于priority为何物?每当插入一个key值时,也会为它对应生成一个随机的priority。这个节点的插入操作分为两步:
- 按BST树的性质插入节点(key, priority)
- 插入完成后,按priority的值去zig-zap,以调整树的平衡性,这样最终生成的树是趋于平衡二叉树的
Treap除去支持insert
, find
, remove
等,还支持快速地查找到最小的priority。于是我们又可以通过STL中的map,set等模拟之。
问题要求
insert(string key, int value)
,插入一个pair,时间复杂度要求为 O(log n)lookup(string key)
, 查询,时间复杂度要求为O(log n)remove(string key)
,删除一个节点,时间复杂度要求为O(log n)max()
,获得当前所有节点中最大的value,时间复杂度要求为O(1)
问题分析
显然,insert
, find
, remove
等方法就可以用BST搞定,比如直接使用STL的map(时间复杂度为O(log n),而STL的unordered_map时间复杂度为O(1)),但是max如何搞定呢?STL的map不支持对所有的value进行维护,比如取得最大值。
那么可以使用两个map+set来搞定,思路如下:
unordered_map<string, int> mp
,用于O(1)的时间来对key进行查找map<int, unordered