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

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

Time Limit: 400 ms Memory Limit: 65536 KiB

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

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

Output
输出平衡二叉树的树根。

Sample Input
5
88 70 61 96 120

Sample Output
70

**Hint
Source
xam**

---

#include <stdio.h>
#include <stdlib.h>
typedef struct node
{
    int data;
    int h;
    struct node *l,*r;
}bitree;

int max(int a,int b)
{
    return a>b ? a:b;//比较大小
}

int deep(bitree *root)//判断树的高度
{
    if(!root) return -1;
    else return root->h;
}

bitree *ll(bitree *root)
{
    bitree *p;
    p=root->l;
    root->l=p->r;
    p->r=root;
    p->h=max(deep(p->l),deep(p->r))+1;
    root->h=max(deep(root->l),deep(root->r))+1;
    return p;
}

bitree *rr(bitree *root)
{
    bitree *p;
    p=root->r;
    root->r=p->l;
    p->l=root;
    p->h=max(deep(p->l),deep(p->r))+1;
    root->h=max(deep(root->l),deep(root->r))+1;
    return p;
}

bitree *lr(bitree *root)
{

    root->l=rr(root->l);//从下往上看，先进行右旋转
    return ll(root);//再进行左旋转
}

bitree *rl(bitree *root)
{

    root->r=ll(root->r);//从下往上看，先进行左旋转
    return rr(root);//再进行右旋转
}
bitree *creat(bitree *root,int m)
{
    if(root==NULL)
    {
        root=(bitree *)malloc(sizeof(bitree));
        root->data=m;
        root->h=0;
        root->l=root->r=NULL;
    }
    else if(m < root->data)
        {
            root->l=creat(root->l,m);
            if(deep(root->l)-deep(root->r)>1)//若左右子树高度差大于1，
            {                                 //要进行旋转
                if(root->l->data > m)
                    root=ll(root);
                else root=lr(root);

            }
        }
        else if(m > root->data)
        {
            root->r=creat(root->r,m);
            if(deep(root->r)-deep(root->l)>1)
            {
                if(root->r->data > m) root=rl(root);
                else root=rr(root);
            }
        }

    root->h=max(deep(root->l),deep(root->r))+1;
    return root;

}
int main()
{
    bitree *root;
    int n,i,m;
    scanf("%d",&n);
    root=NULL;
    for(i=0;i<n;i++)
    {
        scanf("%d",&m);
        root=creat(root,m);
    }
    printf("%d\n",root->data);
    return 0;
}