基于红黑树插入操作原理及java实现方法(分享)

基于红黑树插入操作原理及java实现方法(分享)

红黑树是一种二叉平衡查找树，每个结点上有一个存储位来表示结点的颜色，可以是RED或BLACK。

红黑树具有以下性质：

(1) 每个结点是红色或是黑色

(2) 根结点是黑色的

(3) 如果一个结点是红色的，则它的两个儿子都是黑色的

(4) 对于每个结点，从该结点到其子孙结点的所有路径上包含相同数目的黑结点

通过红黑树的性质，可以保证所有基于红黑树的实现都能保证操作的运行时间为对数级别（范围查找除外。它所需的额外时间和返回的键的数量成正比）。

java的TreeMap就是通过红黑树实现的。

红黑树的操作如果不画图很容易搞糊涂，下面通过图示来说明红黑树的插入操作。

插入一个红色的节点到红黑树中之后，会有6种情况：图示中N表示插入的节点，P表示父节点，U表示叔叔节点，G表示祖父节点，X表示当前操作节点

代码如下：

public class RedBlackBST, Value> {

private Node root;

private static final boolean RED = true;

private static final boolean BLACK = false;

private class Node{

private Key key; //键

private Value val; //值

private Node left, right, parent; //左右子树和父节点

private boolean color; //由其父节点指向它的链接的颜色

public Node(Key key, Value val,Node parent, boolean color){

this.key = key;

this.val = val;

this.color = color;

}

}

public Value get(Key key){

Node x = root;

while(x!=null){

int cmp = key.compareTo(x.key);

if(cmp < 0 ) x = x.left;

else if(cmp > 0) x = x.right;

else return x.val;

}

return null;

}

public void put(Key key, Value val){

if(root==null) { //如果是根节点，就将节点新建为黑色

root = new Node(key,val,null,BLACK);

return;

}

//寻找合适的插入位置

Node parent = null;

Node cur = root;

while(cur!=null) {

parent = cur;

if(key.compareTo(cur.key)>0) cur=cur.right;

else cur = cur.left;

}

Node n = new Node(key,val,parent,RED); //普通的新建节点为红色

//将新节点插入parent下

if(key.compareTo(parent.key) > 0) parent.right = n;

else parent.left = n;

//插入新节点后要调整树中部分节点的颜色和属性来保证红黑树的特征不被破坏

fixAfterInsertion(n);

}

private Node parentOf(Node x) {

return (x==null ? null : x.parent);

}

private boolean colorOf(Node x) {

return (x==null http://3; BLACK : x.color);

}

private Node leftOf(Node x) {

return (x==null ? null : x.left);

}

private Node rightOf(Node x) {

return(x==null ? null : x.right);

}

private void setColor(Node x, boolean color) {

if(x!=null)

x.color = color;

}

private void fixAfterInsertion(Node x) {

while(x!=null && colorOf(parentOf(x)) == RED) {

Node grandPa = parentOf(parentOf(x));

Node parent = parentOf(x);

if(parent == leftOf(grandPa)) {//case 1 || case2 || case3

Node uncle = rightOf(grandPa);

if(colorOf(uncle) == RED) {//case1, uncle is red

setColor(parent,BLACK); //父节点置黑

setColor(uncle, BLACK); //叔叔节点置黑

setColor(grandPa,RED); //祖父节点置红

x = grandPa; //因为祖父节点由黑转红，故要重新调整父节点及其祖先的红黑属性

}else {//case2 || case3,uncle is black

if(x==rightOf(parent)) { //case2

x = parent;

rotateLeft(x);

}

//case3

setColor(parent,BLACK);

setColor(grandPa, RED);

rotateRight(grandPa);

}

}else {//case4 || case 5 || case6

Node uncle = leftOf(grandPa);

if(colorOf(uncle) == RED) { //case4 || case5 || case6

setColor(parent,BLACK);

setColor(uncle, BLACK);

setColor(grandPa,RED);

x = grandPa;

}else{ //case5 || case6, uncle is black

if(x==leftOf(parent)) { //case5

x = parent;

rotateRight(x);

}

//case6

setColor(parent,BLACK);

setColor(grandPa, RED);

rotateLeft(grandPa);

}

}

}

}

private void rotateLeft(Node x) {

if(x==null) return;

Node y = x.right;

x.right = y.left;

if(y.left!=null)

y.left.parent = x;

y.left = x;

y.parent = x.parent;

if(x.parent == null) {

root = y;

}

else if(x.parent.left =wpSbaoB= x) {

x.parentwpSbaoB.left = y;

}else {

x.parent.right = y;

}

x.parent = y;

}

private void rotateRight(Node x) {

if(x==null) return;

Node y = x.left;

x.left = y.right;

if(y.right != null)

y.right.parent = x;

y.right = x;

y.parent = x.parent;

if(x.parent == null) {

root = y;

}else if(x.parent.left==x) {

x.parent.left = y;

}else {

x.parent.right=y;

}

x.parent = y;

}

}

上面的rotateLeft和rotateRight有必要画个图示：

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。