练习4.2 平衡二叉树的根 (25 分)+C语言

练习4.2 平衡二叉树的根 (25 分)

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

输入格式:

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

输出格式:

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

输入样例1:

5
88 70 61 96 120

输出样例1:

70

输入样例2:

7
88 70 61 96 120 90 65

输出样例2:

88

参考答案： 

#include<stdio.h>
#include<stdlib.h>
typedef struct TNode* Tree;
struct TNode{
    int Data;
    Tree Left,Right;
    int Height;
};
typedef Tree AVLTree;
//函数声明
AVLTree SingleLeftRotation(AVLTree T);
AVLTree SingleRightRotation(AVLTree T);
AVLTree DoubleLeftRightRotation(AVLTree T);
AVLTree DoubleRightLeftRotation(AVLTree T);
int Max(int a,int b);
int GetHeight(AVLTree T);
AVLTree Insert(AVLTree T,int data);

int main()
{
    AVLTree A;
    A=NULL;
    int n,data;
    scanf("%d",&n);
    while(n>0){
        scanf("%d",&data);
        A=Insert(A,data);
        n--;
    }
    if(A!=NULL){
        printf("%d",A->Data);
    }
    return 0;
}

int Max(int a,int b){
    return a>b? a:b;//取两者较大值
}
int GetHeight(AVLTree T)
{
    if(T==NULL)
    return 0;
    else
    return T->Height;//获取树的高度便于计算平衡因子
}

AVLTree Insert(AVLTree T,int data){
    if(T==NULL){
        T=(AVLTree)malloc(sizeof(struct TNode));
        T->Data=data;
        T->Left=NULL;
        T->Right=NULL;
        T->Height=1;
    }
    else{
        if(data<T->Data)
        {
            T->Left=Insert(T->Left,data);
            if(GetHeight(T->Left)-GetHeight(T->Right)==2)
            {
                if(data<T->Left->Data){
                    T=SingleLeftRotation(T);//左单旋
                }
                else if(data>T->Left->Data){
                    T=DoubleLeftRightRotation(T);
                }
            }
        }
        else if(data>T->Data)
        {
            T->Right=Insert(T->Right,data);
            if(GetHeight(T->Left)-GetHeight(T->Right)==-2)
            {
                if(data>T->Right->Data){
                    T=SingleRightRotation(T);//右单旋
                }
                else if(data<T->Right->Data){
                    T=DoubleRightLeftRotation(T);
                }
            }
        }
        T->Height=Max(GetHeight(T->Left),GetHeight(T->Right))+1;
    }
    return T;
}
AVLTree SingleLeftRotation(AVLTree T)
{
    AVLTree TL=T->Left;
    T->Left=TL->Right;
    TL->Right=T;
    T->Height=Max(GetHeight(T->Left),GetHeight(T->Right))+1;
    TL->Height=Max(GetHeight(TL->Left),GetHeight(TL->Right))+1;
    return TL;
}
AVLTree SingleRightRotation(AVLTree T)
{
    AVLTree TR=T->Right;
    T->Right=TR->Left;
    TR->Left=T;
    T->Height=Max(GetHeight(T->Left),GetHeight(T->Right))+1;
    TR->Height=Max(GetHeight(TR->Left),GetHeight(TR->Right))+1;
    return TR;
}
AVLTree DoubleLeftRightRotation(AVLTree T){
    T->Left=SingleRightRotation(T->Left);
    return SingleLeftRotation(T);
}
AVLTree DoubleRightLeftRotation(AVLTree T){
    T->Right=SingleLeftRotation(T->Right);
    return SingleRightRotation(T);
}