数据结构实验之查找二:平衡二叉树
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
小小福利:
树的旋转。
#include <iostream>
#include<stdio.h>
#include<stdlib.h>
using namespace std;
typedef struct st
{
struct st *r,*l;
int d,dp;
} tree;
int max(int x,int y)
{
if(x>y)
return x;
else return y;
}
int deep(tree *t)
{
if(t==NULL)
return -1;
else return t->dp;
}
tree *LL(tree *t)//左左情况进行右旋
{
tree *p;
p=t->l;
t->l=p->r;
p->r=t;
p->dp=max(deep(p->l),t->dp)+1;
t->dp=max(deep(t->l),deep(t->r))+1;
return p;
}
tree *RR(tree *t)
{
tree *p;
p=t->r;
t->r=p->l;
p->l=t;
p->dp=max(deep(p->r),t->dp)+1;
t->dp=max(deep(t->l),deep(t->r))+1;
return p;
}
tree *RL(tree *t)
{
t->r=LL(t->r);
return RR(t);
}
tree *LR(tree *t)//左右情况先进行左旋(右右),然后左旋(左左)
{
t->l=RR(t->l);
return LL(t);
}
tree *creat(tree *t,int x)
{
if(t==NULL)
{
t=new tree;
t->l=NULL;
t->r=NULL;
t->d=x;
t->dp=0;
}
else if(x<t->d)
{
t->l=creat(t->l,x);
if(deep(t->l)-deep(t->r)>1)
{
if(x<t->l->d)
{
t=LL(t);
}
else t=LR(t);
}
}
else if(x>t->d)
{
t->r=creat(t->r,x);
if(deep(t->r)-deep(t->l)>1)
{
if(x>t->r->d)
{
t=RR(t);
}
else t=RL(t);
}
}
t->dp=max(deep(t->l),deep(t->r))+1;及时更新每个节点的深度
return t;
}
int main()
{
int i,n,x;
cin>>n;
tree *h;
h=NULL;
for(i=1; i<=n; i++)
{
cin>>x;
h=creat(h,x);
}
cout<<h->d<<endl;
return 0;
}
Source
xam