AVL树的旋转和插入

包含了AVL树的单旋转和双旋转的算法,以及AVL树的递归插入和非递归插入算法。单旋转也叫“一”字形旋转,又可分为左-左型旋转和右-右型旋转;双旋转也叫“之”字形旋转,又可分为左-右型旋转和右-左型旋转。

#include<iostream>
#include<stack>
using namespace std;


class AVLNode{
public:
    int element;
    AVLNode* left;
    AVLNode* right;
    int height;

    AVLNode(int x, AVLNode*l = NULL, AVLNode*r = NULL, int h = 0) :element(x), left(l), right(r), height(h){}

};

int getHeight(AVLNode* t){
    if (t == NULL){
        return -1;
    }
    else{
        int leftHeight = getHeight(t->left);
        int rightHeight = getHeight(t->right);
        return leftHeight > rightHeight ? leftHeight+1 : rightHeight+1;
    }
}


//左-左型旋转
/* K2-> O K2-> O K1-> O / /| / \ K1-> O ----> K1-> O | ----> O O <-K2 / \ / | / O O O O O */
void rotateWithLeftChild(AVLNode* k2){
    AVLNode*k1 = k2->left;
    k2->left = k1->right;
    k1->right = k2;
    k2 = k1;
}
//右-右型旋转
/* K2-> O K2-> O K1-> O \ |\ / \ K1-> O ----> | O <-K1 ----> K2-> O O / \ | \ \ O O O O O */
void rotateWithRightChild(AVLNode* k2){
    AVLNode* k1 = k2->right;
    k2->right = k1->left;
    k1->left = k2;
    k2 = k1;
}

//左-右型双旋转
/* k3-> O k3-> O k3-> O / \ / \ / \ k3->left-> O O O O O O / \ ---> / \ ----> / \ / \ O O K3.left->O O O O O O / \ / \ O O O O */
void doubleWithLeftChild(AVLNode* k3){
    //先对k3的左子树进行右-右型单旋转
    rotateWithRightChild(k3->left);
    //再对k3进行左-左型单旋转
    rotateWithLeftChild(k3);
}

//右-左型双旋转
/* O <-K3 O <-K3 O <-K3 / \ / \ / \ O O <-K3.right O O O O / \ -----> / \ -----> / \ / \ O O O O <-K3.right O O O O / \ / \ O O O O */
void doubleWithRightChild(AVLNode* k3){
    //先对右子树进行左-左型单旋转
    rotateWithLeftChild(k3->right);
    //再对K3进行右-右型单旋转
    rotateWithRightChild(k3);
}

//递归实现AVL树插入
void insert(const int &x, AVLNode* t){
    if (t == NULL){
        t = new AVLNode(x);
    }
    else if (x < t->element){
        insert(x, t->left);
        //如果不平衡就调整
        //在左子树插入,用左高度-右高度
        if (getHeight(t->left) - getHeight(t->right) == 2){
            //如果是左-左型
            if (x < t->left->element){
                //进行单旋转
                rotateWithLeftChild(t);
            }
            //如果是左-右型
            else{
                //进行双旋转
                doubleWithLeftChild(t);
            }
        }
    }
    else if (t->element < x){
        insert(x, t->right);
        //如果不平衡就调整
        //在右子树插入,用右高度-左高度
        if (getHeight(t->right) - getHeight(t->left) == 2){
            //如果是右-右型
            if (x > t->right->element){
                //进行单旋转
                rotateWithRightChild(t);
            }
            //如果是右-左型
            else{
                //进行双调整
                doubleWithRightChild(t);
            }
        }
    }
    else{
        ;
    }

}

//非递归实现AVL树插入
//利用栈来保存每个节点的父节点,实现自底向上检索插入路径上的每个节点是否平衡
void insert_2(const int & x, AVLNode*tree){
    AVLNode * t = tree;
    std::stack<AVLNode*> route;
    //插入
    while (true){
        if (!t){
            t = new AVLNode(x);
            route.push(t);
            break;
        }
        else if (t->element < x){
            route.push(t);
            t = t->right;
            continue;
        }
        else if (x < t->element){
            route.push(t);
            t = t->left;
            continue;
        }
        else{ ; }
    }
    AVLNode *father, *son;

    //调整
    while (true){
        son = route.top();
        route.pop();
        //当栈里只有一个节点时,说明自底向上回溯只剩根节点,则退出循环
        if (route.empty()){
            break;
        }
        father = route.top();
        route.pop();
        //son节点在father节点的左边
        if (son->element < father->element){
            //如果不平衡
            if (getHeight(father->left) - getHeight(father->right) == 2){
                /* 判断插入的x在son节点的左还是右 实质是判断“一”字形还是“之”字形 */
                if (son->element < x){ //x在son节点的右边
                    //说明是“之”字形
                    doubleWithLeftChild(father);
                }
                else if(x < son->element){ //x在son节点的左边
                    //说明是“一”字形
                    rotateWithLeftChild(father);
                }
                else{ //son节点是x
                    //不可能产生不平衡
                    return;
                }
            }
            route.push(father);
        }
        //son节点在father节点的右边
        else{
            //如果不平衡
            if (getHeight(father->right) - getHeight(father->left) == 2){
                if (son->element < x){
                    //"一"字形
                    rotateWithRightChild(father);
                }
                else{
                    doubleWithRightChild(father);
                }
            }
            route.push(father);
        }
    }
}
    原文作者:AVL树
    原文地址: https://blog.csdn.net/DragonBark/article/details/79077782
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞