平衡二叉树

Time Limit: 400MS Memory limit: 65536K

题目描述

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

输入

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

输出

输出平衡二叉树的树根。

示例输入

5
88 70 61 96 120

示例输出

70

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
using namespace std;

struct node
{
int d,h;
node *l,*r;
};
int Max(int a,int b)
{
return a>b?a:b;
}
int dp(node *rt)
{
if(!rt)
return -1;
else
return rt->h;
}
node *LL(node *rt)
{
node *p=rt->l;
rt->l=p->r;
p->r=rt;
rt->h=Max(dp(rt->l),dp(rt->r))+1;
p->h=Max(dp(rt->l),dp(rt->r))+1;
return p;
}
node *RR(node *rt)
{
node *p=rt->r;
rt->r=p->l;
p->l=rt;
rt->h=Max(dp(rt->l),dp(rt->r))+1;
p->h=Max(dp(rt->l),dp(rt->r))+1;
return p;
}
node *LR(node *rt)
{
rt->l=RR(rt->l);
return LL(rt);
}

node *RL(node *rt)
{
rt->r=LL(rt->r);
return RR(rt);
}
node *creat(node *rt,int k)
{
if(!rt)
{
rt=new node;
rt->d=k;
rt->h=0;
rt->l=rt->r=NULL;
}
else
{
if(k<rt->d)
{
rt->l=creat(rt->l,k);
if(dp(rt->l)-dp(rt->r)>1)
{
if(k<rt->l->d)
rt=LL(rt);
else
rt=LR(rt);
}
}
else
{
rt->r=creat(rt->r,k);
if(dp(rt->l)-dp(rt->r)>1)
{
if(k>rt->r->d)
rt=RR(rt);
else
rt=RL(rt);
}

}
}
rt->h=Max(dp(rt->l),dp(rt->r))+1;
return rt;
}
int main()
{
int n,m;
node *head=NULL;
cin>>n;
while(n--)
{
scanf("%d",&m);
head=creat(head,m);
}
cout<<head->d<<endl;
return 0;
}