3374-数据结构实验之查找二：平衡二叉树

#include <bits/stdc++.h>#define ElemType intusing namespace std;typedef struct AVLTree{    ElemType data;    ElemType bal; /// 记录平衡因子    struct AVLTree *Left;    struct AVLTree *Right;}Node;int GetBal(Node *p); /// 求节点平衡因子Node *LL(Node *p); /// 右旋Node *RR(Node *p); /// 左旋Node *LR(Node *p); /// 先右旋再左旋Node *RL(Node *p); /// 先左旋再右旋Node *AVLCreat(Node *root, ElemType x);void InTraversal(Node *root);int main(){    int n;    while(cin >> n)    {        Node *root = NULL; /// 初始化很重要!        for(int i = 0; i < n; i++)        {            int x;            cin >> x;            root = AVLCreat(root, x);        }        cout << root->data << endl;        //InTraversal(root);    }    return 0;}int GetBal(Node *p){    if(!p)    {        return -1;    }    else    {        return p->bal;    }}Node *LL(Node *root) /// 对于LL型 直接在不平衡的节点右旋{    Node *p; /// 画图便于理解    p = root->Left;    root->Left = p->Right;    p->Right = root;    p->bal = max(GetBal(p->Left), GetBal(p->Right)) + 1; /// get平衡因子    root->bal = max(GetBal(root->Left), GetBal(root->Right)) + 1; /// 因为此时节点位置改变 所以重新计算此时节点的平衡因子    return p; /// 此时p为新的子树根节点}Node *RR(Node *root) /// 同上QvQ{    Node *p;    p = root->Right;    root->Right = p->Left;    p->Left = root;    p->bal = max(GetBal(p->Left), GetBal(p->Right)) + 1; /// get平衡因子    root->bal = max(GetBal(root->Left), GetBal(root->Right)) + 1; /// 因为此时节点位置改变 所以重新计算此时节点的平衡因子    return p; /// 此时p为新的子树根节点}Node *LR(Node *p) /// 对于LR型 先对失衡节点的左子树进行左旋 再对整个子树右旋{    p->Left = RR(p->Left); /// 为什么感觉自己不分左右了2333    return LL(p); /// 我真的不分左右了！！！！！！}/// 理解没错 博客有毒Node *RL(Node *p) /// 对于RL型 先对失衡节点的右子树进行右旋 再对整个子树左旋{    p->Right = LL(p->Right);    return RR(p);}Node *AVLCreat(Node *root, ElemType x){    if(!root)    {        root = new Node;        root->data = x;        root->bal = -1;        root->Left = NULL;        root->Right = NULL;    }    else    {        if(x < root->data)        {            root->Left = AVLCreat(root->Left, x);            if(GetBal(root->Left) - GetBal(root->Right) > 1) /// 进入左子树后 只会出现LL或是LR型            {                if(root->Left->data > x) ///则进入左节点的左子树 为LL型                {                    root = LL(root);                }                else                {                    root = LR(root);                }            }        }        else        {            root->Right = AVLCreat(root->Right, x);            if(GetBal(root->Right) - GetBal(root->Left) > 1)            {                if(root->Right->data < x)                {                    root = RR(root);                }                else                {                    root = RL(root);                }            }        }    }    root->bal = max(GetBal(root->Left), GetBal(root->Right)) + 1;    return root;}void InTraversal(Node *root){    if(root)    {        InTraversal(root->Left);        cout << root->data << " ";        InTraversal(root->Right);    }}