数据结构之AVL树

AVL树

平衡二叉树和AVL

平衡二叉树的定义：对于任意一个节点，左子树和右子树的高度差不能为超过1

AVL树：既是平衡二叉树又是二分搜索树的树是AVL树(保持平衡的目的是为了防止二分搜索树退化成链表)

• AVL树是一种最为经典的平衡二叉树
• 满二叉树是一种平衡二叉树
• 完全二叉树是平衡二叉树
• 线段树不是一个完全二叉树，但是一个平衡二叉树

判断是否是二分搜索树

//判断是否是二分搜索树
public boolean isBST() {
    ArrayList<K> keys = new ArrayList<>();
    inOrder(root,keys);
    for(int i = 1; i<keys.size();i++) {
        if(keys.get(i-1).compareTo(keys.get(i)) < 0) {
            return false;
        }
    }
    return true;
}

private void inOrder(Node node, ArrayList<K> keys){
    if(node == null) return;
    inOrder(node.left,keys);
    keys.add(node.key);
    inOrder(node.right,keys);
}

判断是否是平衡二叉树

//判断是否是平衡二叉树
public boolean isBalanced() {
    return isBalanced(root);
}
private boolean isBalanced(Node node) {
    if(node == null)
        return true;
    int balanceFactor = getBalanceFactor(node);
    if(Math.abs(balanceFactor) > 1)
        return false;
    return isBalanced(node.left) && isBalanced(node.right);
}

旋转操作的基本道理

• 插入节点是在不平衡节点的左子节点的左侧对应的是下面左图，这种情况使用右旋转
• 插入节点是在不平衡节点的右子节点的右侧对应的是下面右图，这种情况使用左旋转

右旋转(LL)

当插入的节点导致祖先节点不平衡是由于该插入节点是在该祖先节点的左侧的左侧时

左旋转(RR)

LR和RL

LR

• 先按照下面的方式进行转换
• 然后按照LL的方式(右旋转)进行操作

RL

• 先按照下图的方式进行旋转
• 然后按照RR的方式(左旋转)进行旋转

//向二分搜索树中添加（key,value）,需要考虑平衡维护,共分为4种情况：LL,RR,LR,RL
private Node add(Node node, K key, V value) {
    if(node == null) {
        size ++;
        return new Node(key,value);
    }
    if(key.compareTo(node.key) < 0) {
        node.left = add(node.left,key,value);
    } else if(key.compareTo(node.key) > 0) {
        node.right = add(node.right,key,value);
    } else {
        node.value = value;
    }
    //更新height
    node.height = 1 + Math.max(getHeight(node.left),getHeight(node.right));
    int balanceFactor = getBalanceFactor(node);
    if(Math.abs(balanceFactor) > 1)
        System.out.println("unbanlanced"+balanceFactor);
    //平衡维护
    //LL
    if(balanceFactor > 1 && getBalanceFactor(node.left) >= 0)
        return rightRotate(node);
    //RR
    if(balanceFactor < -1 && getBalanceFactor(node.right) <= 0)
        return leftRotate(node);
    //LR
    if(balanceFactor > 1 && getBalanceFactor(node.left) < 0) {
        node.left = leftRotate(node.left);
        return rightRotate(node);
    }
    //RL
    if(balanceFactor < -1 && getBalanceFactor(node.right) > 0) {
        node.right = rightRotate(node.right);
        return leftRotate(node);
    }
    return node;
}

AVL树的删除