Design Skiplist
Problem
Design a Skiplist without using any built-in libraries.
A skiplist is a data structure that takes O(log(n)) time to add, erase and search. Comparing with treap and red-black tree which has the same function and performance, the code length of Skiplist can be comparatively short and the idea behind Skiplists is just simple linked lists.
For example, we have a Skiplist containing [30,40,50,60,70,90] and we want to add 80 and 45 into it. The Skiplist works this way:
Artyom Kalinin [CC BY-SA 3.0], via Wikimedia Commons
You can see there are many layers in the Skiplist. Each layer is a sorted linked list. With the help of the top layers, add, erase and search can be faster than O(n). It can be proven that the average time complexity for each operation is O(log(n)) and space complexity is O(n).
See more about Skiplist: https://en.wikipedia.org/wiki/Skip_list
Implement the Skiplist class:
Skiplist()Initializes the object of the skiplist.bool search(int target)Returnstrueif the integertargetexists in the Skiplist orfalseotherwise.void add(int num)Inserts the valuenuminto the SkipList.bool erase(int num)Removes the valuenumfrom the Skiplist and returnstrue. Ifnumdoes not exist in the Skiplist, do nothing and returnfalse. If there exist multiplenumvalues, removing any one of them is fine.
Note that duplicates may exist in the Skiplist, your code needs to handle this situation.
Examples
Example 1:
Input
["Skiplist", "add", "add", "add", "search", "add", "search", "erase", "erase", "search"]
[[], [1], [2], [3], [0], [4], [1], [0], [1], [1]]
Output
[null, null, null, null, false, null, true, false, true, false]
Explanation
Skiplist skiplist = new Skiplist();
skiplist.add(1);
skiplist.add(2);
skiplist.add(3);
skiplist.search(0); // return False
skiplist.add(4);
skiplist.search(1); // return True
skiplist.erase(0); // return False, 0 is not in skiplist.
skiplist.erase(1); // return True
skiplist.search(1); // return False, 1 has already been erased.
Solution
A Skiplist is a data structure that allows for fast insertion, deletion, and search by maintaining multiple layers of sorted linked lists. Each higher layer levels up randomly and contains a subset of the elements from the lower level, making searching operations more efficient by skipping over large numbers of elements at once.
Method 1 - Designing with custom node class
Here is the approach:
- Node Class: Define a
Nodeclass to store the value and references to the nodes in different levels. - Skiplist Class:
- Initialization: Set up the Skiplist with a predefined maximum level and a head node that acts as the start of the Skiplist.
- Random Level: Implement a method to randomly determine the level of a new node.
- Search: Implement a method to search for a target value in the Skiplist by traversing from the top level down.
- Add: Implement a method to add a value to the Skiplist, adjusting the levels and pointers accordingly.
- Erase: Implement a method to remove a value from the Skiplist by updating the pointers of the nodes in different levels.
Code
Java
class Skiplist {
class Node {
int val;
Node[] next;
public Node(int val, int size) {
this.val = val;
this.next = new Node[size];
}
}
private static final int MAX_LEVEL = 16;
private final Node head;
private final Random rand;
private int level;
public Skiplist() {
this.head = new Node(-1, MAX_LEVEL);
this.rand = new Random();
this.level = 1;
}
public boolean search(int target) {
Node curr = head;
for (int i = level - 1; i >= 0; i--) {
while (curr.next[i] != null && curr.next[i].val < target) {
curr = curr.next[i];
}
}
curr = curr.next[0];
return curr != null && curr.val == target;
}
public void add(int num) {
Node curr = head;
Node[] update = new Node[MAX_LEVEL];
for (int i = level - 1; i >= 0; i--) {
while (curr.next[i] != null && curr.next[i].val < num) {
curr = curr.next[i];
}
update[i] = curr;
}
int newLevel = randomLevel();
if (newLevel > level) {
for (int i = level; i < newLevel; i++) {
update[i] = head;
}
level = newLevel;
}
Node newNode = new Node(num, newLevel);
for (int i = 0; i < newLevel; i++) {
newNode.next[i] = update[i].next[i];
update[i].next[i] = newNode;
}
}
public boolean erase(int num) {
Node curr = head;
Node[] update = new Node[MAX_LEVEL];
for (int i = level - 1; i >= 0; i--) {
while (curr.next[i] != null && curr.next[i].val < num) {
curr = curr.next[i];
}
update[i] = curr;
}
curr = curr.next[0];
if (curr == null || curr.val != num) {
return false;
}
for (int i = 0; i < level; i++) {
if (update[i].next[i] != curr) {
break;
}
update[i].next[i] = curr.next[i];
}
while (level > 1 && head.next[level - 1] == null) {
level--;
}
return true;
}
private int randomLevel() {
int lvl = 1;
while (lvl < MAX_LEVEL && rand.nextInt(2) == 1) {
lvl++;
}
return lvl;
}
}
Python
class Node:
def __init__(self, val: int, size: int):
self.val = val
self.next = [None] * size
class Skiplist:
MAX_LEVEL = 16
def __init__(self):
self.head = Solution.Node(-1, self.MAX_LEVEL)
self.level = 1
self.random = random.Random()
def search(self, target: int) -> bool:
curr = self.head
for i in range(self.level - 1, -1, -1):
while curr.next[i] and curr.next[i].val < target:
curr = curr.next[i]
curr = curr.next[0]
return curr is not None and curr.val == target
def add(self, num: int) -> None:
update = [None] * self.MAX_LEVEL
curr = self.head
for i in range(self.level - 1, -1, -1):
while curr.next[i] and curr.next[i].val < num:
curr = curr.next[i]
update[i] = curr
new_level = self.random_level()
if new_level > self.level:
for i in range(self.level, new_level):
update[i] = self.head
self.level = new_level
new_node = Solution.Node(num, new_level)
for i in range(new_level):
new_node.next[i] = update[i].next[i]
update[i].next[i] = new_node
def erase(self, num: int) -> bool:
update = [None] * self.MAX_LEVEL
curr = self.head
for i in range(self.level - 1, -1, -1):
while curr.next[i] and curr.next[i].val < num:
curr = curr.next[i]
update[i] = curr
curr = curr.next[0]
if curr is None or curr.val != num:
return False
for i in range(self.level):
if update[i].next[i] != curr:
break
update[i].next[i] = curr.next[i]
while self.level > 1 and self.head.next[self.level - 1] is None:
self.level -= 1
return True
def random_level(self) -> int:
lvl = 1
while lvl < self.MAX_LEVEL and self.random.randint(0, 1):
lvl += 1
return lvl
Complexity
- ⏰ Time complexity: The time complexity for search, add, and erase operations is
O(log(n))on average due to the probabilistic balancing. The worst-case time complexity isO(n), but this happens very rarely due to the random level mechanism. - 🧺 Space complexity:
O(n)wherenis the number of elements in the Skiplist.