数据结构实验之查找二:平衡二叉树
Time Limit: 400ms Memory limit: 65536K 有疑问?点这里^_^
题目描述
根据给定的输入序列建立一棵平衡二叉树,求出建立的平衡二叉树的树根。
输入
输入一组测试数据。数据的第1行给出一个正整数N(n <= 20),N表示输入序列的元素个数;第2行给出N个正整数,按数据给定顺序建立平衡二叉树。
输出
输出平衡二叉树的树根。
示例输入
5 88 70 61 96 120
示例输出
70
提示
平衡二叉树(Balanced Binary Tree)是二叉查找树的一个进化体,也是第一个引入平衡概念的二叉树。1962年,G.M. Adelson-Velsky 和 E.M. Landis发明了这棵树,所以它又叫AVL树。平衡二叉树要求对于每一个节点来说,它的左右子树的高度之差不能超过1,如果插入或者删除一个节点使得高度之差大于1,就要进行节点之间的旋转,将二叉树重新维持在一个平衡状态
。
注意:若在函数LL和RR中没有将p赋值给q,那么会出现断层。因为p是root的引用,在旋转过程中,p不再是跟节点,即root不再是根节点了。
#include<iostream>
using namespace std;
struct node
{
int d;
node *l,*r;
int deep;
};
node *root;
int deept(node *p) //计算树的深度
{
if(!p)
return -1;
return p->deep;
}
void LL(node *&p)
{
//node *tmp=p; //中间变量,因为p是root的引用,交换p和q后,此时root不再是根节点
node *q=p->l;
p->l=q->r;
q->r=p;
p->deep=max(deept(p->r),deept(p->l))+1;
q->deep=max(p->deep,deept(q->l))+1;
p=q; //将p和q交换
//q=tmp;
}
void RR(node *&p)
{
// node *tmp=p;
node *q=p->r;
p->r=q->l;
q->l=p;
p->deep=max(deept(p->l),deept(p->r))+1;
q->deep=max(deept(q->r),p->deep)+1;
p=q;
// q=tmp;
}
void LR(node *&p)
{
RR(p->l);
LL(p);
}
void RL(node *&p)
{
LL(p->r);
RR(p);
}
void InsertT(node *&p,int k)
{
if(!p) //如果节点为空,就在此节点处加入k信息
{
p=new node;
p->d=k;
p->l=p->r=NULL;
p->deep=1;
}
else if(k<p->d) //如果k小于节点的值,就继续在节点的左子树中插入k
{
InsertT(p->l,k);
if(deept(p->l)-deept(p->r)>1) //如果高度之差为2的话就失去了平衡,需要旋转
{
if(k<p->l->d)
LL(p);
else
LR(p);
}
}
else if(k>p->d)
{
InsertT(p->r,k);
if(deept(p->r)-deept(p->l)>1)
{
if(k>p->r->d)
RR(p);
else RL(p);
}
}
p->deep=max(deept(p->l),deept(p->r))+1;
}
int main()
{
int n;
cin>>n;
root=NULL;
while(n--)
{
int x;
cin>>x;
InsertT(root,x);
}
cout<<root->d<<endl;
}