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

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 <bits/stdc++.h>
using namespace std;

typedef struct node{
int data,length;
struct node *l,*r;
}*tree;
int n,m,flag;

int length(tree root)
{
if(root==NULL)
return -1;
else
return root->length;
}

tree RR(tree root)
{
struct node *temp = root->r;
root->r = temp->l;
temp->l = root;
temp->length = max(length(temp->l),length(temp->r))+1;
root->length = max(length(root->l),length(root->r))+1;
return temp;
}
tree LL(tree root)
{

node *temp = root->l;
root->l = temp->r;
temp->r = root;
temp->length = max(length(temp->l),length(temp->r))+1;
root->length = max(length(root->l),length(root->r))+1;
return temp;
}
tree RL(tree root)
{
root->r = LL(root->r);
return RR(root);
}
tree LR(tree root)
{
root->l = RR(root->l);
return LL(root);
}

tree creat(int x,tree root)
{
if(root==NULL)
{
root = new node;
root->data = x;
root->length =0;
root->l = root->r = NULL;
}
else if(x>root->data)
{
root->r = creat(x,root->r);
if(length(root->r)-length(root->l)>1)
{
if(root->r->data<x)
root = RR(root);
else
root = RL(root);
}
}
else if(x<root->data)
{
root->l = creat(x,root->l);
if(length(root->l)-length(root->r)>1)
{
if(root->l->data<x)
root = LR(root);
else
root = LL(root);
}
}
root->length = max(length(root->l),length(root->r))+1;
return root;
}

int main()
{
int x;
cin>>n;
struct node *root = NULL;
for(int i =0; i<n; i++)
{
cin >> x;
root = creat(x,root);
}
cout << root->data <<endl;;
return 0;
}