1. 简介
红黑树是一种近似平衡的二叉搜索树,简单的BST(Binary Search Tree)在没有结构控制的情况下,在某些情况下会表现处不好的性能,如插入序列刚好是一个递增序列,则二叉树的左结点全为空,相当于退化为一个链表。所以BST在理想情况下,大多数时候搜索性能是近似O(log2(n))的, 但在特殊情况下会达到O(n)。为了避免这种情况,就需要保持树的结构,使它较为平衡,从而控制它的深度较小。保证一颗树完全平衡比较麻烦,而红黑树是一种比较适用的近似平衡的二叉树,它有着较好的性能,搜索的时间复杂度为O(log2(n))。
红黑树的使用也比较广泛,jdk1.8中的HashMap中在某一个bin中的链表过长(超过8个,且HashMap的容量大于64时)会将此bin中的链表构建为红黑树,ConcurrentHashMap也一样。TreeSet和TreeMap也是使用红黑树来实现有序性。
2. 实现
记录两种实现,一种是算法第四版中的,基于红链接的实现, 是一个2-3树的等价。一颗大小为N的红黑树的高度不会超过2lgN,所以保证了它的效率。
2.1 基于链接
最好先了解2-3树, 此红黑树定义如下:
1. 红链接均为左链接;
2. 没有任何一个结点同时和两个红链接项链;
3. 任意空链接到根结点的路径上的黑链接的数量相同。(该树是完美黑色平衡的)
这样定义的红黑树等价于一颗2-3树。
2.1.1 Node定义
/**private static final boolean RED = true;
* private static final boolean BLACK = false;
*/
private class Node{
Key key;
Value val;
Node left;
Node right;
int N; //这颗子树中的结点总数, 跟红黑树的实现无关,可以无视
boolean red;//有其父结点指向它的颜色
Node(Key key, Value, val, int N, boolean red){
this.key = key;
this.val = val;
this.N = N;
this.red = red;//
}
private boolean isRed(Node x){
if(x == null) return false;
return x.color == RED;
}
}
根据此红黑树的定义,为了保证其结构定义,可能会用到几种操作:左旋, 右旋, flip()。 (flip():左右链接都为红时,将左右链接置为黑,并将其父结点的链接设为红)
详见 算法第四版 p279
2.1.2 实现:
保持树的平衡性所需的操作是由下往上的,所以只需要在递归调用之后完成这些操作即可。
public class RedBlackBST<Key extends Comparable<Key>, Value>{
private Node root;
private class Node;
private boolean isRed();
private Node rotateLeft(Node h); // 具体实现省略, 返回旋转后处在原来位置的结点引用
private Node rotateRight(Node h); //同上
private void flip(Node h); //
public void put(Key key, Value val){
root = pub(root, key, val);
root.color = BLACK;
}
/**@return 返回插入到树中后的结点的引用*/
private Node put(Node h, Key key, Value val){
if (h == null) //标准插入操作, 和父节点用红链接相连
return new Node(key, val, 1, RED);
int cmp =key.compareTo(h.key);
//递归调用
if(cmp < 0) h.left = put(h.left, key, val);
else if(cmp > 0) h.right = put(h.right, key, val);
else h.val = val;
//递归调用之后,检查树的结构是否满足定义。
if (isRed(h.right) && isRed(h.left)) h = rotateLeft(h);
if(isRed(h.left) && isRed(h.left.left)) h = rotateRight(h);
if(isRed(h.left) && isRed(h.right)) flip(h);
return h;
}
}
2.2 基于红结点
参考:
算法导论:ChA13 红黑树
算法第四版:平衡查找树