Python数据结构与算法之跳表详解
时间：2022-07-16 09:40:48 | 栏目：Python代码 | 点击：次
0. 学习目标
在诸如单链表、双线链表等普通链表中，查找、插入和删除操作由于必须从头结点遍历链表才能找到相关链表，因此时间复杂度均为O(n)。跳表是带有附加指针的链表，使用这些附加指针可以跳过一些中间结点，用以快速完成查找、插入和删除等操作。本节将介绍跳表的相关概念及其具体实现。
通过本节学习，应掌握以下内容：
- 跳表的基本概念及其与普通链表的区别
- 跳表及其操作的实现
1. 跳表的基本概念
跳表 (Skip List) 是一种可以快速进行查找、插入和删除等操作的有序链表(链表中的数据项按照某种顺序进行排列的链表称为有序链表)。跳表的核心思想是以更大的跨度遍历链表，以便可以跳过中间结点。
1.1 跳表介绍
首先，让我们回想一下如何在以下单链表中查找数据元素，即使链表中的数据已经排好序，我们也需要从头结点开始遍历链表。
例如要在以上单链表中查找 15，查找过程为： 0-->3-->5 -->7-->10-->15。
为了提高查找的效率，我们可以抽取链表中的一部分结点建立一个起“索引”作用的链，称为“索引层”或简称“索引”：
此时我们查找 15，首先在索引层中遍历，当我们扫描到值为 10 的结点时，由于下一结点值为 27，因此我们知道 15 应当在这两个结点之间，然后回到原链表中遍历即可找到值为 15 的结点，遍历过程如下图绿色箭头所示：
可以看到通过建立索引链，需要遍历的结点数得以减少，为了进一步减少所需遍历的结点数，可以通过对索引链添加索引，跳表是对链表叠加多级索引的数据结构。
1.2 跳表的性能
1.3 跳表与普通链表的异同
我们可以将跳表理解为排序后的链表，但与普通链表相比，有以下不同点：
- 普通链表中的结点只有一个 next 指针，跳表中的结点有多个 next 指针；
- 给定结点的 next 指针的数量是随机确定的。
跳表中结点的每个 next 指针称为一层，结点中的层数称为结点的大小，其决定了结点在链表中的层级；在普通的有序链表中，插入、删除和查找操作需要对链表依次遍历，因此每个操作的时间复杂度均为O(n)，跳表允许在遍历时跳过一些中间节点，每个操作的时间复杂度为O(㏒n)。
2. 跳表的实现
跳表中的每个数据元素由一个结点表示，结点的层级在结点插入时随机选择，而不考虑数据结构中数据元素的数量。第i级结点有i个next 指针(在结点实现时使用大小为i的数组 next 存储i个 next 指针)，因此我们不需要在结点中存储结点的层级，使用最大层数 max_level 来限制层级上限。跳表的头结点具有从 1 级到 max_level 级的 next 指针。
2.1 跳表结点类
跳表的结点示意图如下所示，每个结点都有一组用于指向其它层级结点的指针域列表 next：
class Node:
def __init__(self, data, level=0):
self.data = data
self.next = [None] * level
def __str__(self):
return "Node({}, {})".format(self.data, len(self.next))
next 数组用于保存指向不同层级结点的指针，其中每个元素就是一个 next 指针，而 data 变量用于存储数据，重载 __str__ 方法用于便于打印结点对象，len(self.next) 可以用于表示结点中的层数。
2.2 跳表的初始化
跳表的初始化是建立一个空的带头结点的跳表，其表长 length 初始化为 0，此时链表中没有元素结点，只有一个头结点：
class SkipList:
def __init__(self, max_level=8):
self.max_level = max_level
node = Node(Node, max_level)
self.head = node
self.length = 0
创建跳表 SkipList 对象的时间复杂度为O(1)。
2.3 获取跳表长度
求取跳表长度只需要重载 __len__ 从对象返回 length 的值，因此时间复杂度为O(1)：
def __len__(self):
return self.length
2.4 读取指定位置元素
由于跳表中的每个结点的层级是在结点插入时随机选择的，因此读取指定位置的数据元素和单链表类似，需要顺序遍历第0层结点，操作的复杂度为O(n)，如果索引不在可接受的索引范围内，将引发 IndexError 异常：：
def __getitem__(self, index):
if index > self.length - 1 or index < 0:
raise IndexError("UnrolledLinkedList assignment index out of range")
else:
count = 0
node = self.head.next[0]
while count < index:
node = node.next[0]
count += 1
return node.data
2.5 查找指定元素
跳表的查找操作从最上层开始，如果待查元素 value 小于当前层的后继结点的 data，则平移至当前层的后继结点；否则下移进入下一层，继续比较，直至第一层。
def locate(self, value):
node = self.head
# 由于索引从0开始，因此第i层级的索引为i-1
level = self.max_level - 1
while level >= 0:
while node.next[level] != None and node.next[level].data < value:
# 平移到当前层的后继结点
node = node.next[level]
# 下移进入下一层
level -= 1
if node.next[0].data == value:
return node.next[0]
else:
raise ValueError("{} is not in skip list".format(value))
2.6 在跳表中插入新元素
与普通链表不同，在跳表中插入新元素不能指定位置(需要保证有序性)，需要通过查找确定新数据元素的插入位置。
插入元素前，首先需要确定该元素所占据的层数 level k，这是通过算法随机生成的，这样可以减少算法实现的复杂度，同时也可以保证跳表性能。
def random_level(self, max_level):
num = random.randint(1, 2**max_level -1)
lognum = math.log(num, 2)
level = int(math.floor(lognum))
return max_level - level
然后需要在 level 1, level 2, ..., level k 各层的链表都插入新元素，为了达到此目的，我们需要在查找插入位置时维护一个列表 update，update 中的每个元素 update[i] 指向 level i 中小于插入元素 data 的最右边结点，即插入位置的左侧，如下图所示，插入元素 data = 9，随机层数 k=3：
def update_list(self, data):
update = [None] * self.max_level
node = self.head
level = self.max_level - 1
while level >= 0:
while node.next[level] != None and node.next[level].data < data:
node = node.next[level]
update[level] = node
level -= 1
return update
可以看到维护 update 列表的过程与查找元素类似，根据 update 列表可以判断跳表中是否已存在元素 data，如果不存在将其插入跳表中：
def insert_node(self, data):
# 产生随机 level，并构造结点
level = self.random_level(self.max_level)
node = Node(data, level)
# 获取 update 列表
update = self.update_list(data)
# 插入结点
if len(update) == 0 or update[0].next[0] == None or update[0].next[0].data != data:
self.length += 1
for i in range(level):
node.next[i] = update[i].next[i]
update[i].next[i] = node
2.7 删除跳表中指定元素
删除指定元素与插入操作类似：
def delete_node(self, data):
# 获取 update 列表
update = self.update_list(data)
if len(update) > 0:
node = update[0].next[0]
# 删除指定元素结点
if node != None and node.data == data:
self.length -= 1
for i in range(len(node.next)):
update[i].next[i] = node.next[i]
del node
2.8 其它一些有用的操作
1.跳表指定层元素输出
将跳表指定层转换为字符串以便进行打印，编写 print_level 方法打印指定层中数据元素：
def get_level(self, level):
"""辅助函数，用于根据给定层构造结果字符串"""
node = self.head.next[level]
result = ''
while node:
result = result + str(node.data) + '-->'
node = node.next[level]
result += 'None'
return result
def print_level(self, level):
print('level {}'.format(level))
result = self.get_level(level)
print(result)
2.跳表各层元素输出
可以借助上述辅助函数 get_level，使用 str 函数调用对象上的 __str__ 方法可以创建适合打印的字符串表示：
def __str__(self):
result = ''
for i in list(range(self.max_level))[self.max_level-1:0:-1]:
result = result + self.get_level(i) + '\n'
result += self.get_level(0)
return result
3. 跳表应用
接下来，我们将测试上述实现的链表，以验证操作的有效性。
3.1 跳表应用示例
首先初始化一个跳表 sl，并在其中插入若干元素：
sl = SkipList(4)
for i in range(0, 30, 3):
sl.insert_node(i)
我们可以直接打印跳表中的数据元素、链表长度等信息：
print('跳表 sl 各层元素为：')
print(sl)
print('跳表 sl 长度为：', len(sl))
print('跳表 sl 第 2 个元素为：', sl[2])
以上代码输出如下：
跳表 sl 各层元素为： 9-->None 0-->9-->18-->None 0-->9-->18-->27-->None 0-->3-->6-->9-->12-->15-->18-->21-->24-->27-->None 跳表 sl 长度为： 10 跳表 sl 第 2 个元素为： 6
接下来，我们将演示在添加/删除元素、以及如何查找指定元素等：
# 插入元素
sl.insert_node(14)
print('在跳表 sl 中插入数据 14 后：')
print(sl)
# 删除元素
sl.delete_node(14)
print('删除跳表 sl 中数据 14 后：')
print(sl)
# 查找数据元素 15
print('查找跳表 sl 中数据元素 15：', sl.locate(15))
以上代码输出如下：
在跳表 sl 中插入数据 14 后： 9-->None 0-->9-->18-->None 0-->9-->14-->18-->27-->None 0-->3-->6-->9-->12-->14-->15-->18-->21-->24-->27-->None 删除跳表 sl 中数据 14 后： 9-->None 0-->9-->18-->None 0-->9-->18-->27-->None 0-->3-->6-->9-->12-->15-->18-->21-->24-->27-->None 查找跳表 sl 中数据元素 15： Node(15, 1)