平衡二叉树

#include <stdio.h>
#include <stdlib.h>
#include <algorithm>
using namespace std;
typedef struct bitnode
{
    int data;
    int d;
    struct bitnode *lchild, *rchild;
}* bitree;

int deep(bitree &t)
{
    if(!t)
        return -1;
    return t->d;
}

bitree LL(bitree &t) //单旋转
{
    bitree q;
    q = t->lchild;
    t->lchild = q->rchild;
    q->rchild = t;
    q->d = max(deep(q->lchild), deep(q->rchild))+1;
    t->d = max(deep(t->lchild), deep(t->rchild))+1;
    return q;
}
bitree RR(bitree &t)//单旋转
{
    bitree q;
    q = t->rchild;
    t->rchild = q->lchild;
    q->lchild = t;
    q->d = max(deep(q->lchild), deep(q->rchild))+1;
    t->d = max(deep(t->lchild), deep(t->rchild))+1;
    return q;
}
bitree LR(bitree &t)//旋转分两步:1.以a为根结点的RR旋转 2.以x为根结点的LL旋转 。
{
    t->lchild = RR(t->lchild);
    return LL(t);
}
bitree RL(bitree &t)//旋转分两步:1.以a为根结点的LL旋转 2.以x为根结点的RR旋转 。
{
    t->rchild = LL(t->rchild);
    return RR(t);
}

bitree Insert(bitree &t, int x)
{
    if(!t)
    {
        t = new bitnode;
        t->lchild = NULL;
        t->rchild = NULL;
        t->data = x;
        t->d = 0;
    }
    else if(x<t->data)
    {
        t->lchild = Insert(t->lchild, x);
        if(deep(t->lchild)-deep(t->rchild)>1)
        {
            if(x<t->lchild->data)
                t = LL(t);
            else
                t = LR(t);
        }
    }
    else if(x>t->data)
    {
        t->rchild = Insert(t->rchild, x);
        if(deep(t->rchild)-deep(t->lchild)>1)
        {
            if(x>t->rchild->data)
                t = RR(t);
            else
                t = RL(t);
        }
    }

    t->d = max(deep(t->lchild), deep(t->rchild))+1;
    return t;
}
int main()
{
    int n, i, num;
    scanf("%d", &n);
    bitree t;
    t = NULL;
    for(i=0; i<n; i++)
    {
        scanf("%d", &num);
        Insert(t, num);
    }
    printf("%d\n", t->data);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

StriveBen

写字不易,给点动力吧

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值