为何Redis使用跳表,而非红黑树实现SortedSet?
1 什么是跳表
William Pugh发明,他在论文《Skip lists: a probabilistic alternative to balanced trees》中详细介绍了跳表的数据结构和插入删除等操作,论文是这么介绍跳表的:
Skip lists are a data structure that can be used in place of balanced trees.Skip lists use probabilistic balancing rather than strictly enforced balancing and as a result the algorithms for insertion and deletion in skip lists are much simpler and significantly faster than equivalent algorithms for balanced trees.
即跳表可用来替代红黑树,使用概率均衡技术,使得插入、删除操作更简单、更快。先来看论文里的一张图:

- a:已排好序的链表,查找一个结点最多需比较N个结点
- b:每隔2个结点增加一个指针,指向该结点间距为2的后续结点,查找一个结点最多比较ceil(N/2)+1个结点
- c,每隔4个结点增加一个指针,指向该结点间距为4的后续结点,查找一个结点最多比较ceil(N/4)+1个结点
- 若每第
2^i个结点都有一个指向间距为2^i的后续结点的指针,不断增加指针,比较次数降为log(N)。搜索会很快,但插入、删除很困难
一个拥有k个指针的结点称为一个k层结点(level k node)。按上述逻辑,50%的结点为1层,25%的结点为2层,12.5%的结点为3层…若每个结点的层数随机选取,是否仍服从这样的分布呢?
使一个k层结点的第i个指针指向第i层的下一个结点,而非它后面的第2^(i-1)个结点,则结点的插入和删除只需原地修改操作;一个结点的层数,是在它被插入时随机选取的,并且永不改变。因为这样的数据结构是基于链表的,并且额外的指针会跳过中间结点,所以称为跳表(Skip Lists)。
二分查找底层依赖数组随机访问的特性,所以只能用数组实现。若数据存储在链表,就无法用二分搜索了?
只需稍微改造下链表,就能支持类似“二分”搜索,即跳表(Skip list),支持快速的新增、删除、搜索操作。
2 Redis有序集合(Sorted Set)
就是用跳表实现。红黑树也能实现快速插入、删除和查找操作。Redis为何不选择红黑树?
单链表,即使存储数据有序,若搜索某数据,也只能从头到尾遍历,搜索效率低,平均时间复杂度O(n)。
3 提高链表结构搜索效率
若如下图,对链表建立一级“索引”,每两个结点提取一个结点到上一级,把抽出来的那级叫作索引或索引层。图中的down表示down指针,指向下一级结点。

如搜索16:
- 先遍历索引层,当遍历到索引层的13时,发现下一个结点是17,说明目标结点位于这俩结点中间
- 然后通过down指针,下降到原始链表层,继续遍历
此时只需再遍历2个结点,即可找到16!
原先单链表结构需遍历10个结点,现在只需遍历7个结点。加一层索引,所需遍历的结点个数的确减少了,搜索效率提升!
再加层索引,搜索效率会更高吗?每两个结点再抽出一个结点到第二级索引。现在搜索16,只需遍历6个结点:

数据量大一点,有一64结点链表,给它建立五级索引:

原来没有索引时,单链表搜索62需遍历62个结点!现在只需遍历11个。当链表长度n很大,建立索引后,搜索性能显著提升。这种有多级索引的,可提高查询效率的链表就是跳表。
4 跳表的搜索时间复杂度
单链表搜索时间复杂度O(n),跳表呢?
若链表有n个结点,有多少级索引?假设每两个结点抽出一个结点作为上级索引,则:
- 第一级索引结点个数是n/2
- 第二级n/4
- 第三级n/8
- …
- 第k级就是
n/(2^k)
假设索引有h级,最高级索引有2个结点,可得:
n/(2h) = 2
所以:
h = log2n-1
若包含原始链表这一层,整个跳表的高度就是log2N。在跳表中查询某数据时,若每层都要遍历m个结点,那在跳表中查询一个数据的时间复杂度就是O(m*logn)。
那这m值多少?按前面这种索引结构,每级索引都最多只需遍历3个结点,即m=3
5 为何是3?
假设要查找数据x,在第k级索引中,遍历到y结点后,发现x大于y,小于后面的结点z,所以通过y的down指针,从第k级索引下降到第k-1级索引。在第k-1级索引中,y和z之间只有3个结点(包含y和z),所以,我们在K-1级索引中最多只需要遍历3个结点,依次类推,每一级索引都最多只需要遍历3个结点。
所以跳表中查询任意数据时间复杂度就是O(logn)。跟二分查找一样。其实是基于单链表实现了二分查找,这种查询效率的提升,前提是建立了很多级索引,也就是我们在第6节讲过的空间换时间的设计思路。
6 跳表内存占用
由于跳表要存储多级索引,势必比单链表消耗更多存储空间。那到底是多少呢?若原始链表大小为n:
- 第一级索引大约有n/2个结点
- 第二级索引大约有n/4个结点
- …
- 最后一级2个结点
多级结点数总和:
n/2+n/4+n/8…+8+4+2=n-2
所以空间复杂度是O(n)。量还是挺大,能否再稍微
7 索引占用内存空间优化
若每三五个结点才抽取一个到上级索引:

- 第一级索引需要大约n/3个结点
- 第二级索引需要大约n/9个结点
- 每往上一级,索引结点个数都除以3
假设最高级索引结点个数为1,总索引结点数:
n/3+n/9+n/27+…+9+3+1=n/2
尽管空间复杂度还是O(n),但比上面的每两个结点抽一个结点的索引构建方法,要减少一半的索引结点存储空间。
大可不必过分在意索引占用额外空间,实际开发中,原始链表中存储的有可能是很大的对象,而索引结点只需存储关键值和几个指针,无需存储对象,所以当对象比索引结点大很多时,那索引占用的额外空间可忽略。
8 插入和删除的时间复杂度
8.1 插入
插入一个数据,O(logn)时间复杂度。
单链表中,一旦定位好要插入位置,插入时间复杂度O(1)。但为保证原始链表中数据的有序性,要先找到插入位置,所以查找操作较耗时。
纯单链表,需遍历每个结点以找到插入位置。但跳表搜索某结点的的时间复杂度O(logn),所以搜索某数据应插入的位置的时间复杂度也是O(logn):

8.2 删除
如果这个结点在索引中也有出现,除了要删除原始链表的结点,还要删除索引中的。
因为单链表删除操作需拿到要删除结点的前驱结点,然后通过指针完成删除。所以查找要删除结点时,一定要获取前驱结点。若是双向链表,就没这个问题了。
9 索引动态更新
当不停往跳表插入数据时,若不更新索引,可能出现某2个索引结点之间数据非常多。极端时退化成单链表。

需维护索引与原始链表大小之间的平衡,即若链表结点多,索引结点就相应增加,避免复杂度退化及查找、插入、删除操作性能下降。
像红黑树、AVL树等平衡二叉树通过左、右旋保持左右子树平衡,而跳表通过随机函数维护“平衡性”。
往跳表插入数据时,可选择同时将这个数据插入到部分索引层中。
9.1 如何选择加入哪些索引层?
通过随机函数,决定将这个结点插入到哪几级索引。如随机函数生成值K,就把这个结点添加到第一级到第K级索引。

10 为何Redis用跳表来实现有序集合,而非红黑树?
10.1 Redis有序集合核心操作
- 插入一个数据
- 删除一个数据
- 查找一个数据
- 迭代输出有序序列
以上操作,红黑树也能完成,时间复杂度跟跳表一样。
- 按照区间查找数据
红黑树效率低于跳表。跳表可做到O(logn)定位区间的起点,然后在原始链表顺序往后遍历。
除了性能的其它原因
- 代码实现比红黑树好懂、好写多了,因为简单就代表可读性好,不易出错
- 跳表更灵活,可通过改变索引构建策略,有效平衡执行效率和内存消耗
因为红黑树比跳表诞生更早,很多编程语言中的Map类型(比如JDK 的 HashMap)都是通过红黑树实现的。业务开发时,直接从JDK拿来用,但跳表没有一个现成的实现,只能自己实现。
11 跳表实现(Java 版)
① 数据结构
- 表中的元素使用结点表示,结点的层数在它被插入时随机计算决定(与表中已有结点数目无关)
- 一个i层的结点有i个前向指针(java中使用结点对象数组forward来表示),索引为从1~i
- MaxLevel记录跳表的最大层数
- 跳表的层数,为当前所有结点中的最大层数(若list为空,则层数为1)
- 列表头header拥有从1到MaxLevel的前向指针
public class SkipList<T> {
// 最高层数
private final int MAX_LEVEL;
// 当前层数
private int listLevel;
// 表头
private SkipListNode<T> listHead;
// 表尾
private SkipListNode<T> NIL;
// 生成randomLevel用到的概率值
private final double P;
// 论文里给出的最佳概率值
private static final double OPTIMAL_P = 0.25;
public SkipList() {
// 0.25, 15
this(OPTIMAL_P, (int)Math.ceil(Math.log(Integer.MAX_VALUE) / Math.log(1 / OPTIMAL_P)) - 1);
}
public SkipList(double probability, int maxLevel) {
P = probability;
MAX_LEVEL = maxLevel;
listLevel = 1;
listHead = new SkipListNode<T>(Integer.MIN_VALUE, null, maxLevel);
NIL = new SkipListNode<T>(Integer.MAX_VALUE, null, maxLevel);
for (int i = listHead.forward.length - 1; i >= 0; i--) {
listHead.forward[i] = NIL;
}
}
// 内部类
class SkipListNode<T> {
int key;
T value;
SkipListNode[] forward;
public SkipListNode(int key, T value, int level) {
this.key = key;
this.value = value;
this.forward = new SkipListNode[level];
}
}
}
② 搜索算法
按key搜索,找到返回该key对应的value,未找到则返回null。
通过遍历forward数组,找特定的searchKey。假设skip list的key从小到大排列,从跳表当前最高层listLevel开始寻找searchKey。
在某层找到一个不<searchKey的结点后,跳到下层继续找,直到最底层
根据最后搜索停止位置的下一个结点,就可以判断searchKey在不在跳表中。
跳表中查找8

③ 插入和删除算法
通过查找与连接(search and splice):

维护一个update数组,在搜索结束之后,update[i]保存的是待插入/删除结点在第i层的左侧结点。
插入
若key不存在,则插入该key与对应的value;若key存在,则更新value。
如果待插入的结点的层数高于跳表的当前层数listLevel,则更新listLevel。
选择待插入结点的层数randomLevel:
randomLevel只依赖于跳表的最高层数和概率值p。
另一种实现方法为,如果生成的randomLevel大于当前跳表的层数listLevel,那么将randomLevel设置为listLevel+1,这样方便以后的查找,在工程上是可以接受的,但同时也破坏了算法的随机性。
删除
删除特定的key与对应的value。如果待删除的结点为跳表中层数最高的结点,那么删除之后,要更新listLevel。
public class SkipList<T> {
// 最高层数
private final int MAX_LEVEL;
// 当前层数
private int listLevel;
// 表头
private SkipListNode<T> listHead;
// 表尾
private SkipListNode<T> NIL;
// 生成randomLevel用到的概率值
private final double P;
// 论文里给出的最佳概率值
private static final double OPTIMAL_P = 0.25;
public SkipList() {
// 0.25, 15
this(OPTIMAL_P, (int)Math.ceil(Math.log(Integer.MAX_VALUE) / Math.log(1 / OPTIMAL_P)) - 1);
}
public SkipList(double probability, int maxLevel) {
P = probability;
MAX_LEVEL = maxLevel;
listLevel = 1;
listHead = new SkipListNode<T>(Integer.MIN_VALUE, null, maxLevel);
NIL = new SkipListNode<T>(Integer.MAX_VALUE, null, maxLevel);
for (int i = listHead.forward.length - 1; i >= 0; i--) {
listHead.forward[i] = NIL;
}
}
// 内部类
class SkipListNode<T> {
int key;
T value;
SkipListNode[] forward;
public SkipListNode(int key, T value, int level) {
this.key = key;
this.value = value;
this.forward = new SkipListNode[level];
}
}
public T search(int searchKey) {
SkipListNode<T> curNode = listHead;
for (int i = listLevel; i > 0; i--) {
while (curNode.forward[i].key < searchKey) {
curNode = curNode.forward[i];
}
}
if (curNode.key == searchKey) {
return curNode.value;
} else {
return null;
}
}
public void insert(int searchKey, T newValue) {
SkipListNode<T>[] update = new SkipListNode[MAX_LEVEL];
SkipListNode<T> curNode = listHead;
for (int i = listLevel - 1; i >= 0; i--) {
while (curNode.forward[i].key < searchKey) {
curNode = curNode.forward[i];
}
// curNode.key < searchKey <= curNode.forward[i].key
update[i] = curNode;
}
curNode = curNode.forward[0];
if (curNode.key == searchKey) {
curNode.value = newValue;
} else {
int lvl = randomLevel();
if (listLevel < lvl) {
for (int i = listLevel; i < lvl; i++) {
update[i] = listHead;
}
listLevel = lvl;
}
SkipListNode<T> newNode = new SkipListNode<T>(searchKey, newValue, lvl);
for (int i = 0; i < lvl; i++) {
newNode.forward[i] = update[i].forward[i];
update[i].forward[i] = newNode;
}
}
}
public void delete(int searchKey) {
SkipListNode<T>[] update = new SkipListNode[MAX_LEVEL];
SkipListNode<T> curNode = listHead;
for (int i = listLevel - 1; i >= 0; i--) {
while (curNode.forward[i].key < searchKey) {
curNode = curNode.forward[i];
}
// curNode.key < searchKey <= curNode.forward[i].key
update[i] = curNode;
}
curNode = curNode.forward[0];
if (curNode.key == searchKey) {
for (int i = 0; i < listLevel; i++) {
if (update[i].forward[i] != curNode) {
break;
}
update[i].forward[i] = curNode.forward[i];
}
while (listLevel > 0 && listHead.forward[listLevel - 1] == NIL) {
listLevel--;
}
}
}
private int randomLevel() {
int lvl = 1;
while (lvl < MAX_LEVEL && Math.random() < P) {
lvl++;
}
return lvl;
}
public void print() {
for (int i = listLevel - 1; i >= 0; i--) {
SkipListNode<T> curNode = listHead.forward[i];
while (curNode != NIL) {
System.out.print(curNode.key + "->");
curNode = curNode.forward[i];
}
System.out.println("NIL");
}
}
public static void main(String[] args) {
SkipList<Integer> sl = new SkipList<Integer>();
sl.insert(20, 20);
sl.insert(5, 5);
sl.insert(10, 10);
sl.insert(1, 1);
sl.insert(100, 100);
sl.insert(80, 80);
sl.insert(60, 60);
sl.insert(30, 30);
sl.print();
System.out.println("---");
sl.delete(20);
sl.delete(100);
sl.print();
}
}
更多推荐


所有评论(0)