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

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

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

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

Sample Input
5
88 70 61 96 120
Sample Output
70

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

struct node
{
    int data;
    int deep;
    struct node *left, *right;
};

int max(int x, int y)
{
    if(x>y) return x;
    else return y;
}

int deep(struct node *root)
{
    if(root==NULL) return 0;
    else return root->deep;
}

struct node *ll(struct node *root)
{
    struct node *p, *q;
    p = root;
    q = root->left;
    p->left = q->right;
    q->right = p;
    p->deep = max(deep(p->left), deep(p->right))+1;
    q->deep = max(deep(q->left), deep(q->right))+1;
    return q;
};

struct node *rr(struct node *root)
{
    struct node *p, *q;
    p = root;
    q = root->right;
    p->right = q->left;
    q->left = p;
    p->deep = max(deep(p->right), deep(p->left))+1;
    q->deep = max(deep(q->right), deep(q->left))+1;
    return q;
};

struct node *rl(struct node *root)
{
    root->right = ll(root->right);
    root = rr(root);
    return root;
}

struct node *lr(struct node *root)
{
    root->left = rr(root->left);
    root = ll(root);
    return root;
}

struct node *add(struct node *root, int x)
{
    if(root==NULL)
    {
        root = (struct node *)malloc(sizeof(struct node));
        root->left = NULL;
        root->right = NULL;
        root->data = x;
        root->deep = 1;
    }
    else
    {
        if(x<root->data)
        {
            root->left = add(root->left, x);
            if(deep(root->left)-deep(root->right) > 1)
            {
                if(x < root->left->data) root = ll(root);
                else root = lr(root);
            }
        }
        else
        {
            root->right = add(root->right, x);
            if(deep(root->right)-deep(root->left) > 1)
            {
                if(x < root->right->data) root = rl(root);
                else root = rr(root);
            }
        }
        root->deep = max(deep(root->left), deep(root->right)) + 1;
    }
    return root;
}

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