平衡二叉树
-
平衡查找树(AVL树)
AVL树是一种左右子树高度之差不超过1,且左右子树内部也符合高度差不超过1的性质。
下图有节点的左右子树高度差都超过了1,所有不是AVL树。
平衡因子: 某结点的左子树与右子树的高度(深度)差即为该结点的平衡因子(BF,Balance Factor)。平衡二叉树上所有结点的平衡因子只可能是 -1,0 或 1。如果某一结点的平衡因子绝对值大于1则说明此树不是平衡二叉树。为了方便计算每一结点的平衡因子我们可以为每个节点赋予height这一属性,表示此节点的高度。
最小不平衡子树:距离插入节点最近的,且平衡因子的绝对值大于1的节点为根的子树。
引起不平衡的情况
在AVL树中插入删除节点可能会导致不平衡情况产生,分为以下四种。
- 对平衡树中A节点的左子树的左孩子插入一个节点(LL)
- 对平衡树中A节点的右子树的右孩子插入一个节点(RR)
- 对平衡树中A节点的左子树的右孩子插入一个节点(LR)
- 对平衡树中A节点的右子树的左孩子插入一个节点(RL)
下面用图的方式给出这四种情况怎么旋转
AVL树代码,节点的实现比较简单代码与二叉树节点代码一样只是增加了height属性,如下。
节点结构
//省略了get/set方法和构造函数
public class AVLNode <K extends Comparable<K>>{
public int height;
public AVLNode parent;
public AVLNode left;
public AVLNode right;
public K key;
}
工具类代码
//获取子树高度
private int GetHeight(AVLNode avlNode){
if (avlNode == null)
return 0;
else
return avlNode.getHeight();
}
//设置子树高度
private int SetHeight(AVLNode a,AVLNode b){
return GetHeight(a) > GetHeight(b) ? GetHeight(a) + 1 : GetHeight(b) + 1;
}
//前序遍历
void PreOrder(AVLNode avlNode){
if (avlNode == null){
return;
}
System.out.println(avlNode.getKey());
PreOrder(avlNode.getLeft());
PreOrder(avlNode.getRight());
}
AVL树的平衡就是通过一次或者多次左旋右旋进行调整。
左旋右旋
/**
* 左旋的步骤
* 第一步获取旋转节点p的右孩子pr,将pr的左孩子rl设为p的右孩子。
* 第二步将p的父节点设置为p右孩子pr的父节点,此时有两种情况,p的父节点
* 不存在那么直接将pr设置为root节点;p的父节点存在,查看p是父节点的
* 左孩子还是右孩子,将pr设置为对应的孩子。
* 第三步将p设置为pr的左孩子,pr设置为p的父节点。
* p p pr
* /\ / \ \ /\
* pl pr ===> pl rl pr ===> p rr
* /\ \ /\
* rl rr rr pl rl
* */
private AVLNode left_change(AVLNode avlNode) {
AVLNode lnode = avlNode.right;
avlNode.right = lnode.left;
if (avlNode.right != null){
avlNode.right.parent = avlNode;
}
lnode.parent = avlNode.parent;
if (avlNode.parent != null){
if (avlNode == avlNode.parent.left){
avlNode.parent.left = lnode;
} else {
avlNode