数据结构_练习4.2 平衡二叉树的根_AVLtree

• 题目地址
• 题目分析：左左和右右直接处理，左右改左左，有右左改右右。

#include<stdio.h>
#include<stdlib.h>
#define max(a,b) ((a)>(b)?(a):(b))

typedef struct tree *tp;
struct tree{
    int data;
    tp left,right;
};

tp inAVL(tp rootp,tp tmp);
int  getHight(tp tmp){
    if(tmp==NULL) return 0;
    if(tmp->left==NULL&&tmp->right==NULL) return 1;

    return max(getHight(tmp->left),getHight(tmp->right))+1;
}
tp leftRota(tp tmp){
    tp record=tmp->left;
    tmp->left=tmp->left->right;
    record->right=tmp;
    
    return record;
}
tp rightRota(tp tmp){
    tp record=tmp->right;
    tmp->right=tmp->right->left;
    record->left=tmp;
    
    return record;
}

int main()
{
    int n;
    scanf("%d",&n);

    tp rootp=NULL;
    for(int i=0,data;i<n;i++)
    {
        scanf("%d",&data);
        tp tmp=(tp)malloc(sizeof(tp));
        tmp->data=data,tmp->left=NULL,tmp->right=NULL;
        rootp=inAVL(rootp,tmp);
    }
    printf("%d",rootp->data);  
    
    return 0;
}
tp inAVL(tp rootp,tp tmp)
{
    if(rootp==NULL) return tmp;

    if(tmp->data < rootp->data) rootp->left=inAVL(rootp->left,tmp);
    else rootp->right=inAVL(rootp->right,tmp);

    int factor=getHight(rootp->left)-getHight(rootp->right);
    if(factor==2)
    {
        if(tmp->data > rootp->left->data)       //不平衡在左边的右边
            rootp->left=rightRota(rootp->left); //子部右旋，形成左左

        rootp=leftRota(rootp);
    }
    else if(factor==-2)
    {
        if(tmp->data < rootp->right->data)
            rootp->right=leftRota(rootp->right);

        rootp=rightRota(rootp);
    }

    return rootp;
}