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

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

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

题目链接：

http://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Contest/contestproblem/cid/2712/pid/3374

#include <bits/stdc++.h>

using namespace std;

typedef struct treenode{
    int date;
    int depth;
    struct treenode *leftnode;
    struct treenode *rightnode;
}node;

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

node *LL(node *root)
{
    node *p;
    p=root->leftnode;
    root->leftnode=p->rightnode;
    p->rightnode=root;
    p->depth=max(deep(p->leftnode),deep(p->rightnode))+1;
    root->depth=max(deep(root->leftnode),deep(root->rightnode))+1;
    return p;
}

node *RR(node *root)
{
    node *p;
    p=root->rightnode;
    root->rightnode=p->leftnode;
    p->leftnode=root;
    p->depth=max(deep(p->leftnode),deep(p->rightnode))+1;
    root->depth=max(deep(root->leftnode),deep(root->rightnode))+1;
    return p;
}

node *LR(node *root)
{
    root->leftnode=RR(root->leftnode);
    root=LL(root);
    return root;
}

node *RL(node *root)
{
    root->rightnode=LL(root->rightnode);
    root=RR(root);
    return root;
}

node *create(node *root,int x)
{
    if(!root)
    {
        root=new node;
        root->date=x;
        root->depth=0;
        root->leftnode=NULL;
        root->rightnode=NULL;
    }
    else
    {
        if(x<root->date)
        {
            root->leftnode=create(root->leftnode,x);
            if(deep(root->leftnode)-deep(root->rightnode)>1)
            {
                if(x<root->leftnode->date)
                root=LL(root);
                else
                root=LR(root);
            }
        }
        else
        {
            root->rightnode=create(root->rightnode,x);
            if(deep(root->rightnode)-deep(root->leftnode)>1)
            {
                if(x>root->rightnode->date)
                root=RR(root);
                else
                root=RL(root);
            }
        }
    }
    root->depth=max(deep(root->leftnode),deep(root->rightnode))+1;
    return root;
}

int main()
{
    int n;
    node *root;
    while(cin >> n)
    {
        root=NULL;
        while(n--)
        {
            int x;
            cin >> x;
            root=create(root,x);
        }
        cout << root->date << endl;
    }
    return 0;
}