平衡二叉树

创立平衡二叉树,插入数据后旋转使其平衡,返回根结点

mooc-10
旋转操作原理参考
代码参考(有注释)

#include <stdio.h>
#include <stdlib.h>
typedef struct AVLNode *AVLTree;
struct AVLNode
{
    int data;
    AVLTree left;
    AVLTree right;
    int height; //树高
};
int Max(int a,int b)
{
    return a>b?a:b;
}
int getHeight(AVLTree a)
{
    return a==NULL?-1:a->height;
}
AVLTree LLRotation(AVLTree A)
{
    AVLTree B=A->left;
    A->left=B->right;
    B->right=A;
    A->height=Max(getHeight(A->left),getHeight(A->right))+1;
    B->height=Max(getHeight(B->left),A->height)+1;
    return B;
}
AVLTree RRRotation(AVLTree A)
{
    AVLTree B=A->right;
    A->right=B->left;
    B->left=A;
    A->height=Max(getHeight(A->left),getHeight(A->right))+1;
    B->height=Max(getHeight(B->left),A->height)+1;
    return B;
}
AVLTree LRRotation(AVLTree A)
{
    A->left=RRRotation(A->left);
    return LLRotation(A);
}
AVLTree RLRotation(AVLTree A)
{
    A->right=LLRotation(A->right);
    return RRRotation(A);
}
AVLTree insert(AVLTree T,int data)
{
    if(!T)
    {
        T=(AVLTree)malloc(sizeof(struct AVLNode));
        T->data=data;
        T->left=NULL;
        T->right=NULL;
        T->height=0;
    }
    else
    {
        if(data<T->data)
        {
            T->left=insert(T->left,data);
            if((getHeight(T->left)-getHeight(T->right))>1)
               {
                   if(data<T->left->data) T=LLRotation(T);
                    else if(data>T->left->data) T=LRRotation(T);
               }
        }
        else if(data>T->data)
        {
            T->right=insert(T->right,data);
            if((getHeight(T->right)-getHeight(T->left))>1)
               {
                   if(data<T->right->data) T=RLRotation(T);
                    else if(data>T->right->data) T=RRRotation(T);
               }
        }
    }
    T->height=Max(getHeight(T->left),getHeight(T->right))+1;
    return T;
}
int main()
{
    int n;
    scanf("%d",&n); //有多少个结点
    AVLTree T=NULL;
    for(int i=0;i<n;i++)
    {
        int num;
        scanf("%d",&num); //结点数据
        T=insert(T,num);
    }
    printf("%d\n",T->data);
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值