红黑树的简单插入

1. p,u遍黑色，z变红色
2. p变黑色，z变红色，以p为轴，pz为半径右旋。
3. 以x为轴，px为半径左旋（x向上走），得到二图情形。

class Node {
	int data;
	boolean isRed;//是红节点还是黑节点
	Node parent;
	Node leftChiled;
	Node rightChiled;
	public Node(int data) {
		this.data=data;
		isRed=true;
	}
}
public class RedBlackTree {
	private static Node root;
	public void insert(int id) {
		Node newNode=new Node(id);
		if(root==null) {
			root=newNode;
			root.parent=null;
			root.isRed=false;//根节点总是黑色，规则2
		} else {
			Node curParent=root;
			Node current=root;
			while(true) {
				curParent=current;
				if(id<current.data) {
					current=current.leftChiled;
					if(current==null) {
						current=newNode;
						curParent.leftChiled=current;
						current.parent=curParent;
						break;
					}
				} else {
					current=current.rightChiled;
					if(current==null) {
						current=newNode;
						curParent.rightChiled=current;
						current.parent=curParent;
						break;
					}
				}
			}
			//判断是否平衡，变色旋转
			insertFix(current);
		}
	}

	public void insertFix(Node node) {
		while(node.parent!=null&& node.parent.isRed ) {
			if(node.parent==node.parent.parent.leftChiled) {	//左子节点
				//判断叔叔节点是黑色还是红色
				Node y=node.parent.parent.rightChiled;
				if(y.isRed) {
					//父节点，祖父节点，叔叔节点都变成相反颜色
					node.parent.isRed=false;
					node.parent.parent.isRed=true;
					node.parent.parent.rightChiled.isRed=false;
					node=node.parent.parent;
				} else if(node==node.parent.rightChiled) {//右子树
					node=node.parent;
					leftRotate(node);
				} else {
					//父节点，祖父节点变色
					node.parent.isRed=false;
					node.parent.parent.isRed=true;
					//node=node.parent;
					rightRotate(node.parent.parent);
				}
			} else {//父节点是祖父节点的右子节点
				//判断叔叔节点是黑色还是红色
				Node y=node.parent.parent.leftChiled;
				if(y.isRed) {
					//父节点，祖父节点，叔叔节点都变成相反颜色
					node.parent.isRed=false;
					node.parent.parent.isRed=true;
					node.parent.parent.leftChiled.isRed=false;
					node=node.parent.parent;
				} else if(node==node.parent.leftChiled) {//右子树
					node=node.parent;
					rightRotate(node);
				} else {
					//父节点，祖父节点变色
					node.parent.isRed=false;
					node.parent.parent.isRed=true;
					//node=node.parent;
					leftRotate(node.parent.parent);
				}
			}
		}
		root.isRed=false;
	}
	//以x为轴进行右旋
	public void leftRotate(Node x) {
		if(x==null) return ;
		Node y=x.leftChiled;//x的左子节点
		x.leftChiled=y.rightChiled;
		if(y.rightChiled!=null) {
			y.leftChiled.parent=x;
		}
		y.parent=x.parent;
		if(x.parent==null) {//x为根节点
			root=y;
		} else if(x==x.parent.leftChiled) {
			y.parent.leftChiled=y;
		} else {
			y.parent.rightChiled=y;
		}
		y.rightChiled=x;
		x.parent=y;
	}
	//以x为轴进行左旋
	public void rightRotate(Node x) {
		if(x==null) return ;
		Node y=x.rightChiled;
		x.rightChiled=y.leftChiled;
		if(y.rightChiled!=null) {
			y.rightChiled.parent=x;
		}
		y.parent=x.parent;
		if(x.parent==null) {
			root=y;
		} else if(x.parent.leftChiled==x) {
			y.parent.leftChiled=y;
		} else {
			y.parent.rightChiled=y;
		}
		y.leftChiled=x;
		x.parent=y;
	}
	/**
	 * 中序遍历
	 * @param node
	 */
	public void orderTraversal(Node root) {
		if(root!=null) {
			orderTraversal(root.leftChiled);
			System.out.print(root.data+" ");
			orderTraversal(root.rightChiled);
		}
	}
	public static void main(String[] args) {
		RedBlackTree rbt=new RedBlackTree();
		int[] a=new int[]{50,25,75,12,37,62,87,6,18,31,43};
		for(int i=0;i<a.length;i++) {
			rbt.insert(a[i]);
		}
		rbt.orderTraversal(root);
	}
}