ConcurrentHashMap红黑树的实现
红黑树
红黑树是一种特殊的二叉树,主要用它存储有序的数据,提供高效的数据检索,时间复杂度为O(lgn),每个节点都有一个标识位表示颜色,红色或黑色,有如下5种特性:
1、每个节点要么红色,要么是黑色;
2、根节点一定是黑色的;
3、每个空叶子节点必须是黑色的;
4、如果一个节点是红色的,那么它的子节点必须是黑色的;
5、从一个节点到该节点的子孙节点的所有路径包含相同个数的黑色节点;
注:首先它是二叉树,所以还是要满足:左节点hash值<父节点<右节点
结构示意图
只要满足以上5个特性的二叉树都是红黑树,当有新的节点加入时,有可能会破坏其中一些特性,需要通过左旋或右旋操作调整树结构,重新着色,使之重新满足所有特性。
ConcurrentHashMap红黑树实现
在1.8的实现中,当一个链表中的元素达到8个时,会调用treeifyBin()
方法把链表结构转化成红黑树结构,实现如下:
1 /** 2 * Replaces all linked nodes in bin at given index unless table is 3 * too small, in which case resizes instead. 4 */ 5 private final void treeifyBin(Node<K,V>[] tab, int index) { 6 Node<K,V> b; int n, sc; 7 if (tab != null) { 8 if ((n = tab.length) < MIN_TREEIFY_CAPACITY) 9 tryPresize(n << 1); 10 else if ((b = tabAt(tab, index)) != null && b.hash >= 0) { 11 synchronized (b) { 12 if (tabAt(tab, index) == b) { 13 TreeNode<K,V> hd = null, tl = null; 14 for (Node<K,V> e = b; e != null; e = e.next) { 15 TreeNode<K,V> p = 16 new TreeNode<K,V>(e.hash, e.key, e.val, 17 null, null); 18 if ((p.prev = tl) == null) 19 hd = p; 20 else 21 tl.next = p; 22 tl = p; 23 } 24 setTabAt(tab, index, new TreeBin<K,V>(hd)); 25 } 26 } 27 } 28 } 29 }
从上述实现可以看出:并非一开始就创建红黑树结构,如果当前Node
数组长度小于阈值MIN_TREEIFY_CAPACITY
,默认为64,先通过扩大数组容量为原来的两倍以缓解单个链表元素过大的性能问题。
红黑树构造过程
下面对红黑树的构造过程进行分析:
1、通过遍历Node
链表,生成对应的TreeNode
链表,其中TreeNode
在实现上继承了Node
类;
1 class TreeNode<K,V> extends Node<K,V> { 2 TreeNode<K,V> parent; // red-black tree links 3 TreeNode<K,V> left; 4 TreeNode<K,V> right; 5 TreeNode<K,V> prev; 6 // needed to unlink next upon deletion 7 boolean red; 8 }
假设TreeNode
链表如下,其中节点中的数值代表hash
值:
2、根据TreeNode
链表初始化TreeBin
类对象,TreeBin
在实现上同样继承了Node
类,所以初始化完成的TreeBin
类对象可以保持在Node
数组中;
1 class TreeBin<K,V> extends Node<K,V> { 2 TreeNode<K,V> root; 3 volatile TreeNode<K,V> first; 4 volatile Thread waiter; //volatile修饰,内存可见, 5 volatile int lockState; 6 // values for lockState 7 // 持有写锁时,设置为1 8 static final int WRITER = 1; 9 // 等待写锁时的设置为2 10 static final int WAITER = 2; 11 // increment value for setting read lock 12 static final int READER = 4; 13 }
3、初始化根节点:
遍历TreeNode
链表生成红黑树,一开始二叉树的根节点root
为空,则设置链表中的第一个节点80为root
,并设置其red
属性为false
,因为在红黑树的特性1中,明确规定根节点必须是黑色的;
1 for (TreeNode<K,V> x = b, next; x != null; x = next) { 2 next = (TreeNode<K,V>)x.next; 3 x.left = x.right = null; 4 if (r == null) { 5 x.parent = null; 6 x.red = false; 7 r = x; 8 } 9 ...
二叉树结构:
4、加入节点60,如果root
不为空,则通过比较节点hash
值的大小将新节点插入到指定位置,实现如下:
1 K k = x.key; 2 int h = x.hash; 3 Class<?> kc = null;//泛型,泛型擦除在编译期间 4 for (TreeNode<K,V> p = r;;) { 5 int dir, ph; 6 K pk = p.key; 7 if ((ph = p.hash) > h) 8 dir = -1; 9 else if (ph < h) 10 dir = 1; 11 else if ((kc == null && 12 (kc = comparableClassFor(k)) == null) || 13 (dir = compareComparables(kc, k, pk)) == 0) 14 dir = tieBreakOrder(k, pk); 15 TreeNode<K,V> xp = p; 16 if ((p = (dir <= 0) ? p.left : p.right) == null) { 17 x.parent = xp; 18 if (dir <= 0) 19 xp.left = x; 20 else 21 xp.right = x; 22 r = balanceInsertion(r, x); 23 break; 24 } 25 }
其中x
代表即将插入到红黑树的节点,p
指向红黑树中当前遍历到的节点,从根节点开始递归遍历,x
的插入过程(根据hash值的大小,找到插入位置并插入)如下:
1)、如果x
的hash
值小于p
的hash
值,则判断p
的左节点是否为空,如果不为空,则把p
指向其左节点,并继续和p
进行比较,如果p
的左节点为空,则把x
指向的节点插入到该位置;
2)、如果x
的hash
值大于p
的hash
值,则判断p
的右节点是否为空,如果不为空,则把p
指向其右节点,并继续和p
进行比较,如果p
的右节点为空,则把x
指向的节点插入到该位置;
3)、如果x
的hash
值和p
的hash
值相等,怎么办?
解决:首先判断节点中的key
对象的类是否实现了Comparable
接口,如果实现Comparable
接口,则调用compareTo
方法比较两者key
的大小,但是如果key
对象没有实现Comparable
接口,或则compareTo
方法返回了0,则继续调用tieBreakOrder
方法计算dir
值,tieBreakOrder
方法实现如下:
static int tieBreakOrder(Object a, Object b) {
int d;
if (a == null || b == null ||
(d = a.getClass().getName().
compareTo(b.getClass().getName())) == 0)
d = (System.identityHashCode(a) <= System.identityHashCode(b) ?
-1 : 1);
return d;
}
最终比较key
对象的默认hashCode()
方法的返回值,因为System.identityHashCode(a)
调用的是对象a
默认的hashCode()
;
插入节点60之后的二叉树:
5、当有新节点加入时,可能会破坏红黑树的特性,需要执行balanceInsertion()
方法调整二叉树,使之重新满足特性,方法中的变量xp
指向x
的父节点,xpp
指向xp
父节点,xppl
和xppr
分别指向xpp
的左右子节点,balanceInsertion()
方法首先会把新加入的节点设置成红色。
①、加入节点60之后,此时xp
指向节点80,其父节点为空,直接返回。
if ((xp = x.parent) == null) {
x.red = false;
return x;
}
else if (!xp.red || (xpp = xp.parent) == null)
return root;
调整之后的二叉树:
②、加入节点50,二叉树如下:
继续执行balanceInsertion()
方法调整二叉树,此时节点50的父节点60是左儿子,走如下逻辑:
if (xp == (xppl = xpp.left)) {
if ((xppr = xpp.right) != null && xppr.red) {
xppr.red = false;
xp.red = false;
xpp.red = true;
x = xpp;
}
else {
if (x == xp.right) {
root = rotateLeft(root, x = xp);
xpp = (xp = x.parent) == null ? null : xp.parent;
}
if (xp != null) {
xp.red = false;
if (xpp != null) {
xpp.red = true;
root = rotateRight(root, xpp);
}
}
}
}
根据上述逻辑,把节点60设置成黑色,把节点80设置成红色,并对节点80执行右旋操作,右旋实现如下:
static <K,V> TreeNode<K,V> rotateRight(TreeNode<K,V> root,
TreeNode<K,V> p) {
TreeNode<K,V> l, pp, lr;
if (p != null && (l = p.left) != null) {
if ((lr = p.left = l.right) != null)
lr.parent = p;
if ((pp = l.parent = p.parent) == null)
(root = l).red = false;
else if (pp.right == p)
pp.right = l;
else
pp.left = l;
l.right = p;
p.parent = l;
}
return root;
}
右旋之后的红黑树如下:
③、加入节点70,二叉树如下:
继续执行balanceInsertion()
方法调整二叉树,此时父节点80是个右儿子,节点70是左儿子,且叔节点50不为空,且是红色的,则执行如下逻辑:
if (xppl != null && xppl.red) {
xppl.red = false;
xp.red = false;
xpp.red = true;
x = xpp;
}
此时二叉树如下:
此时x
指向xpp
,即节点60,继续循环处理x
,设置其颜色为黑色,最终二叉树如下:
④、加入节点20,二叉树变化如下:
因为节点20的父节点50是一个黑色的节点,不需要进行调整;
⑤、加入节点65,二叉树变化如下:
对节点80进行右旋操作。
⑥、加入节点40,二叉树变化如下:
1、对节点20执行左旋操作;
2、对节点50执行右旋操作;
最后加入节点10,二叉树变化如下:
重新对节点进行着色,到此为止,红黑树已经构造完成;