定义
跳表是可以实现二分查找的有序链表
抛硬币
插入的时候有一点需要注意,就是在原始链表里插入一个数之后,要不要去更新索引,因为如果不更新索引,那么一旦插入的数据比较多,那么跳表存在的意义还是不大。但若是每插入一次就更新又没有必要。所以这里用抛硬币来决定要不要添加索引。从下往上,每一次都通过抛硬币来决定要不要在这一层添加一个索引。
题目
1206-设计跳表
不使用任何库函数,设计一个跳表。
跳表是在 O(log(n)) 时间内完成增加、删除、搜索操作的数据结构。跳表相比于树堆与红黑树,其功能与性能相当,并且跳表的代码长度相较下更短,其设计思想与链表相似。
例如,一个跳表包含 [30, 40, 50, 60, 70, 90],然后增加 80、45 到跳表中,以下图的方式操作:
跳表中有很多层,每一层是一个短的链表。在第一层的作用下,增加、删除和搜索操作的时间复杂度不超过 O(n)。跳表的每一个操作的平均时间复杂度是 O(log(n)),空间复杂度是 O(n)。
在本题中,你的设计应该要包含这些函数:
- bool search(int target) : 返回target是否存在于跳表中。
- void add(int num): 插入一个元素到跳表。
- bool erase(int num): 在跳表中删除一个值,如果 num 不存在,直接返回false. 如果存在多个 num ,删除其中任意一个即可。
注意,跳表中可能存在多个相同的值,你的代码需要处理这种情况。
代码
class Skiplist {
static class Node {
int val;
Node right, down;
public Node(Node r, Node d, int val) {
right = r;
down = d;
this.val = val;
}
}
Node head = new Node(null, null, 0);
public boolean search(int target) {
for (Node p = head; p != null; p = p.down) {
while (p.right != null && p.right.val < target) {
p = p.right;
}
if (p.right != null && p.right.val == target) {
return true;
}
}
return false;
}
// 2^64 已经相当大了,这里换成栈来操作会超时,不知道为啥
Node[] stack = new Node[64];
public void add(int num) {
int lv = -1;
for (Node p = head; p != null; p = p.down) {
while (p.right != null && p.right.val < num) {
p = p.right;
}
stack[++lv] = p;
}
boolean insertUp = true;
Node downNode = null;
while (insertUp && lv >= 0) {
Node insert = stack[lv--];
insert.right = new Node(insert.right, downNode, num);
downNode = insert.right;
insertUp = (Math.random()<0.5);
}
if (insertUp) {
head = new Node(new Node(null, downNode, num), head, 0);
}
}
public boolean erase(int num) {
boolean exists = false;
for (Node p = head; p != null; p = p.down) {
while (p.right != null && p.right.val < num) {
p = p.right;
}
if (p.right != null && p.right.val == num) {
exists = true;
p.right = p.right.right;
}
}
return exists;
}
}
/**
* Your Skiplist object will be instantiated and called as such:
* Skiplist obj = new Skiplist();
* boolean param_1 = obj.search(target);
* obj.add(num);
* boolean param_3 = obj.erase(num);
*/