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

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

Time Limit: 400MS  Memory Limit: 65536KB

Submit  Statistic  Discuss

Problem Description

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

Input

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

Output

输出平衡二叉树的树根。

Example Input

5
88 70 61 96 120

Example Output

70

Hint

 

Author

xam 

参考1：http://blog.csdn.net/u010442302/article/details/52713585

参考2：http://blog.csdn.net/flying_star521/article/details/52311038

参考3：http://blog.csdn.net/annfan123/article/details/52201273

#include <bits/stdc++.h>
using namespace std;
#define Max(a,b)(a>b?a:b)
typedef struct node
{
    node*left,*right;
    int data;
    int h;
} tree;
int geth(tree*root)///注意设立一个获得高度的函数,且只要是和获得高度相关的操作都要用，不要直接调用高度
{
    if(!root)
        return 0;
    else
        return root->h;
}
tree*ll(tree*root)
{
    tree*q=root->left;
    root->left=q->right;
    q->right=root;
    root->h=Max(geth(root->left),geth(root->right))+1;///注意返回的高度+1
    q->h=Max(geth(q->left),geth(q->right))+1;
    return q;
}
tree*rr(tree*root)
{
    tree*q=root->right;
    root->right=q->left;
    q->left=root;
    root->h=Max(geth(root->left),geth(root->right))+1;
    q->h=Max(geth(q->left),geth(q->right))+1;
    return q;
}
tree*lr(tree*root)
{
    root->left=rr(root->left);///注意不要写成rr(root->left);否则WA
    return ll(root);
}
tree*rl(tree*root)
{
    root->right=ll(root->right);
    return rr(root);
}
tree*creat(tree*root,int x)
{
    if(root==NULL)
    {
        root=new tree;
        root->left=root->right=NULL;///为了规范和易读尽量记得写上，不过不写此题也会AC
        root->data=x;
        root->h=1;
    }
    else if(x<root->data)
    {
        root->left=creat(root->left,x);
        if(geth(root->left)-geth(root->right)>1)
        {
            if(x<root->left->data)///注意单旋和双旋的判断条件
                root=ll(root);///这是带返回值的函数,注意一定不要只写ll(root)，下同(或者return ll(root))
            else
                root=lr(root);
        }
    }
    else if(root->data<x)
    {
        root->right=creat(root->right,x);
        if(geth(root->right)-geth(root->left)>1)
        {
            if(x>root->right->data)
                root=rr(root);
            else
                root=rl(root);
        }
    }
    root->h=Max(geth(root->left),geth(root->right))+1;///记得每次插入点后要让高度提升(?)
    return root;
}
int main()
{
    ///std::ios::sync_with_stdio(false);
    int n,x;
    tree*root=NULL;
    scanf("%d",&n);
    for(int i=0; i<n; i++)
    {
        scanf("%d",&x);
        root=creat(root,x);
    }
    printf("%d\n",root->data);
    return 0;
}