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

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

Time Limit: 400MS Memory Limit: 65536KB

Problem Description

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

Input

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

Output

输出平衡二叉树的树根。

Example Input

5
88 70 61 96 120

Example Output

70

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
struct node
{
    int data;
    int deep;
    struct node *left, *right;
};

int tdeep(struct node *root)
{
    if(!root)
        return -1;
    else
        return root->deep;
}

struct node *ll(struct node *root)
{
    struct node *temp;
    temp=root->left;
    root->left=temp->right;
    temp->right=root;
    temp->deep= tdeep(temp->left) >tdeep(temp->right)? tdeep(temp->left)+1 :tdeep(temp->right)+1;
    root->deep= tdeep(root->left) >tdeep(root->right)? tdeep(root->left)+1 :tdeep(root->right)+1;
    return temp;
}

struct node *rr(struct node *root)
{
    struct node *temp;
    temp=root->right;
    root->right=temp->left;
    temp->left=root;
    temp->deep= tdeep(temp->left) >tdeep(temp->right)? tdeep(temp->left)+1 :tdeep(temp->right)+1;
    root->deep= tdeep(root->left) >tdeep(root->right)? tdeep(root->left)+1 :tdeep(root->right)+1;
    return temp;
}

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

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

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

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