数据结构实验之查找二:平衡二叉树
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
**Hint
Source
xam**
#include <stdio.h>
#include <stdlib.h>
typedef struct node
{
int data;
int h;
struct node *l,*r;
}bitree;
int max(int a,int b)
{
return a>b ? a:b;//比较大小
}
int deep(bitree *root)//判断树的高度
{
if(!root) return -1;
else return root->h;
}
bitree *ll(bitree *root)
{
bitree *p;
p=root->l;
root->l=p->r;
p->r=root;
p->h=max(deep(p->l),deep(p->r))+1;
root->h=max(deep(root->l),deep(root->r))+1;
return p;
}
bitree *rr(bitree *root)
{
bitree *p;
p=root->r;
root->r=p->l;
p->l=root;
p->h=max(deep(p->l),deep(p->r))+1;
root->h=max(deep(root->l),deep(root->r))+1;
return p;
}
bitree *lr(bitree *root)
{
root->l=rr(root->l);//从下往上看,先进行右旋转
return ll(root);//再进行左旋转
}
bitree *rl(bitree *root)
{
root->r=ll(root->r);//从下往上看,先进行左旋转
return rr(root);//再进行右旋转
}
bitree *creat(bitree *root,int m)
{
if(root==NULL)
{
root=(bitree *)malloc(sizeof(bitree));
root->data=m;
root->h=0;
root->l=root->r=NULL;
}
else if(m < root->data)
{
root->l=creat(root->l,m);
if(deep(root->l)-deep(root->r)>1)//若左右子树高度差大于1,
{ //要进行旋转
if(root->l->data > m)
root=ll(root);
else root=lr(root);
}
}
else if(m > root->data)
{
root->r=creat(root->r,m);
if(deep(root->r)-deep(root->l)>1)
{
if(root->r->data > m) root=rl(root);
else root=rr(root);
}
}
root->h=max(deep(root->l),deep(root->r))+1;
return root;
}
int main()
{
bitree *root;
int n,i,m;
scanf("%d",&n);
root=NULL;
for(i=0;i<n;i++)
{
scanf("%d",&m);
root=creat(root,m);
}
printf("%d\n",root->data);
return 0;
}