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

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

Time Limit: 400ms   Memory limit: 65536K  有疑问？点这里^_^

题目描述

根据给定的输入序列建立一棵平衡二叉树，求出建立的平衡二叉树的树根。

输入

输入一组测试数据。数据的第1行给出一个正整数N(n <= 20)，N表示输入序列的元素个数；第2行给出N个正整数，按数据给定顺序建立平衡二叉树。

输出

输出平衡二叉树的树根。

示例输入

5
88 70 61 96 120

示例输出

70

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int data,bf;
    node *L,*R;
};
node *Creat()
{
    node *p=new node;
    p->bf=0;
    p->L=NULL;
    p->R=NULL;
    return p;
}
int n;
int Height(node *&p)
{
    if(p==NULL)
    return -1;
    return p->bf;
}
node *LLrotate(node *p)
{
    node *p1;
    p1=p->L;
    p->L=p1->R;
    p1->R=p;
    p->bf=max(Height(p->L),Height(p->R))+1;
    p1->bf=max(Height(p1->L),Height(p1->R))+1;
    return p1;
}
node *RRrotate(node *p)
{
    node *p1;
    p1=p->R;
    p->R=p1->L;
    p1->L=p;
    p->bf=max(Height(p->L),Height(p->R))+1;
    p1->bf=max(Height(p1->L),Height(p1->R))+1;
    return p1;
}
node *LRrotate(node *p)
{
    p->L=RRrotate(p->L);
    return LLrotate(p);
}
node *RLrotate(node *p)
{
    p->R=LLrotate(p->R);
    return RRrotate(p);
}
void AVLinsert(node *&p,int elem)
{
    if(p==NULL)
    {
        p=Creat();
        p->data=elem;
        return ;
    }
    if(elem<p->data)
    {
        AVLinsert(p->L,elem);
        if(Height(p->L)-Height(p->R)==2)
        {
            if(elem<p->L->data)
                p=LLrotate(p);
            else
                p=LRrotate(p);
        }
    }
    if(elem>p->data)
    {
        AVLinsert(p->R,elem);
        if(Height(p->R)-Height(p->L)==2)
        {
            if(elem>p->R->data)
             p=RRrotate(p);
             else
            p=RLrotate(p);
        }
    }
    p->bf=max(Height(p->L),Height(p->R))+1;
}

int main()
{
    while(~scanf("%d",&n))
    {
        node *root=Creat();
        root=NULL;
         for(int i=0;i<n;i++)
    {
        int t;
        scanf("%d",&t);
         AVLinsert(root,t);
    }
        printf("%d\n",root->data);
    }
    return 0;
}