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

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

Description

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

Input

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

Output

输出平衡二叉树的树根。

Sample

Input 

5
88 70 61 96 120

Output 

70

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
typedef struct node
{
    int c, d;
    struct node *lt, *rt;
}*tree;
int deep(tree root)
{
    if(root)
        return root->d;
    return -1;
}
int max1(int a, int b)
{
    if(a>b)
        return a;
    return b;
}
tree LL(tree root)
{
    tree q;
    q = root->lt;
    root->lt = q->rt;
    q->rt = root;
    root->d = max1(deep(root->lt), deep(root->rt))+1;
    return q;
}
tree RR(tree root)
{
    tree q;
    q = root->rt;
    root->rt = q->lt;
    q->lt = root;
    root->d = max1(deep(root->lt), deep(root->rt))+1;
    return q;
}
tree LR(tree root)
{
    root->lt = RR(root->lt);
    root = LL(root);
    return root;
}
tree RL(tree root)
{
    root->rt = LL(root->rt);
    root = RR(root);
    return root;
}
tree add(tree root, int x)
{
    if(root==NULL)
    {
        root = (tree)malloc(sizeof(struct node));
        root->c = x;
        root->d = 0;
        root->lt = NULL;
        root->rt = NULL;
    }
    else
    {
        if(x<root->c)
        {
            root->lt = add(root->lt, x);
            if(deep(root->lt)-deep(root->rt)>1)
            {
                if(x<root->lt->c)
                    root = LL(root);
                else
                    root = LR(root);
            }
        }
        else
        {
            root->rt = add(root->rt, x);
            if(deep(root->rt)-deep(root->lt)>1)
            {
                if(x>root->rt->c)
                    root = RR(root);
                else
                    root = RL(root);
            }
        }
    }
    root->d = max1(deep(root->lt), deep(root->rt))+1;
    return root;
}
int main()
{
    int n, i;
    tree root;
    scanf("%d", &n);
    root = NULL;
    for(i = 0;i<n;i++)
    {
        int x;
        scanf("%d", &x);
        root = add(root, x);
    }
    printf("%d\n", root->c);
    return 0;
}