练习4.2 平衡二叉树的根

题目

将给定的一系列数字插入初始为空的AVL树，请你输出最后生成的AVL树的根结点的值。

输入格式:
输入的第一行给出一个正整数N（≤20），随后一行给出N个不同的整数，其间以空格分隔。

输出格式:
在一行中输出顺序插入上述整数到一棵初始为空的AVL树后，该树的根结点的值。

输入样例1:

5
88 70 61 96 120

输出样例1:

70

推荐大家先看一下小灰的漫画算法2里对AVL树的讲解，个人觉得还不错。
传送门

过程分析

以 3 2 1 4 5 6 7 10 9 8为例
3，2可以正常地插入树中，但是到了插入1的时候，根结点3的平衡因子变成了2，形成了左左局面，得以3为根节点进行右旋转（顺时针旋转），结果2成了根节点，3成了2的右孩子

平衡因子是指一个节点的左子树与右子树的高度差

继续增加节点4，没什么问题。

增加节点5时，形成了右右局面，节点3的平衡因子变为-1，得以节点3为根节点进行左旋（逆时针旋转）

一直这样子增加，直到插入节点9，红圈一块形成了右左局面，得先对10右旋，再对7左旋

这个例子没有讲左右局面，补一个。
先对L左旋，再对P右旋。

代码

#include <stdio.h>

#define max(a, b) ((a) > (b) ? (a) : (b))
typedef struct BiTNode {
    int data;
    struct BiTNode *leftChild;
    struct BiTNode *rightChild;
} BiTNode, *BiTree;

int getHeight(BiTree T) {
    return (!T) ? 0 : max(getHeight(T->leftChild), getHeight(T->rightChild)) + 1;
}

/*针对左左局面做右旋转（顺时针旋转）*/
void leftLeft(BiTree *T) {
    BiTree L;
    L = (*T)->leftChild;  // L指向T的左子树根节点 
    (*T)->leftChild = L->rightChild; // L的右子树挂接为T的左子树
    L->rightChild = (*T);
    (*T) = L; // T指向新的根节点
}
/*针对右右局面做左旋转（逆时针旋转）*/
void rightRight(BiTree *T) {
    BiTree R;
    R = (*T)->rightChild;  // R指向T的右子树根节点
    (*T)->rightChild = R->leftChild;  // L的右子树挂接为T的左子树
    R->leftChild = (*T);
    (*T) = R;   // T指向新的根节点
}
/*针对左右局面做双旋*/
void leftRight(BiTree *T) {
    rightRight(&(*T)->leftChild); // 对T的左孩子进行左旋转（可以把T的左孩子看作处于相对的右右局面（不是真正的右右局面））
    leftLeft(T); // 上一步操作后，树变成了左左局面
}
/*针对右左局面做双旋*/
void rightLeft(BiTree *T) {
    leftLeft(&(*T)->rightChild); // 对T的右孩子进行右旋转
    rightRight(T); // 上一步操作后，树变成了右右局面
}

void insert(BiTree *T, int data) {
    if (!(*T)) {
        (*T) = (BiTree)malloc(sizeof(BiTNode));
        (*T)->data = data;
        (*T)->leftChild = NULL;
        (*T)->rightChild = NULL;
    } else {
        if (data > (*T)->data) { 
            insert(&(*T)->rightChild, data);
            if (getHeight((*T)->rightChild) - getHeight((*T)->leftChild) == 2) {
                // 一开始我还以为data就是一定等于T的右孩子的data呀。
                // 但是上一步insert后，data就不是T的右孩子的data,即使insert的参数的T的右孩子；data已经是T的右孩子的左右孩子中的一个data。因为上一步insert已经跳出了，树的结构改变了
                // 自己简单走一遍递归流程就能理解的
                // 如果data大于T的右孩子的data，说明带这个data的节点变成了T的右孩子的右孩子，此时的树是右右局面
                // 如果data小于T的右孩子的data，说明带这个data的节点变成了T的右孩子的左孩子，此时的树是右左局面
                data > (*T)->rightChild->data ? rightRight(T) : rightLeft(T);
            }
        } else {
            insert(&(*T)->leftChild, data);
            if (getHeight((*T)->leftChild) - getHeight((*T)->rightChild) == 2) {
                data < (*T)->leftChild->data ? leftLeft(T) : leftRight(T);
            }
        }
    }
}

int main(){
    BiTree T=NULL;
    int count,data;
    scanf("%d",&count);
    while(count--){
        scanf("%d",&data);
        insert(&T,data);
    }
    printf("%d",T->data);
    
    return 0;
}