题目描述
根据给定的输入序列建立一棵平衡二叉树,求出建立的平衡二叉树的树根。
输入
输入一组测试数据。数据的第1行给出一个正整数N(n <= 20),N表示输入序列的元素个数;第2行给出N个正整数,按数据给定顺序建立平衡二叉树。
输出
输出平衡二叉树的树根。
示例输入
5 88 70 61 96 120
示例输出
70
提示
一定要记住每一步的回溯,搞清楚根节点是什么,根据二叉树的左右子树的高度差来判断进行什么样的调整
1.调整方法
(1)插入点位置必须满足二叉查找树的性质,即任意一棵子树的左结点都小于根结点,右结点大于根结点
(2)找出插入结点后不平衡的最小二叉树进行调整,如果是整个树不平衡,才进行整个树的调整。
2.调整方式
(1)LL型
LL型:插入位置为左子树的左结点,进行向右旋转
由于在A的左孩子B的左子树上插入结点F,使A的平衡因子由1变为2,成为不平衡的最小二叉树根结点。此时A结点顺时针右旋转,旋转过程中遵循“旋转优先”的规则,A结点替换D结点成为B结点的右子树,D结点成为A结点的左孩子。
(2)RR型
RR型:插入位置
为右子树的右孩子,进行向左旋转
由于在A的右子树C的右子树插入了结点F,A的平衡因子由-1变为-2,成为不平衡的最小二叉树根结点。此时,A结点逆时针左旋转,遵循“旋转优先”的规则,A结点替换D结点成为C的左子树,D结点成为A的右子树。
(
(3)LR型
LR型:插入位置为
左子树的右孩子,要进行两次旋转,先左旋转(RR型),再右旋转(LL型);第一次最小不平衡子树的根结点先不动,调整插入结点所在的子树,第二次再调整最小不平衡子树。
(4)RL型
RL型:插入位置为右子树的左孩子,进行两次调整,先右旋转再左旋转;处理情况与LR类似。
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
typedef int Elemtype;
typedef struct node
{
Elemtype data;
int d;
struct node *lchild,*rchild;
};
int max(int x,int y)
{
return x>y ? x:y;
}
int Deep(struct node *head)
{
if (!head)
return -1;
else return head->d;
}
struct node *LL(struct node *head)//右旋
{
struct node *p = head->lchild;
head->lchild = p->rchild;
p->rchild = head;
p->d = max(Deep(p->lchild),Deep(p->rchild))+1;
head->d = max(Deep(head->lchild),Deep(head->rchild))+1;
return p;
}
struct node *RR(struct node *head)//左旋
{
struct node *p = head->rchild;
head->rchild = p->lchild;
p->lchild = head;
p->d = max(Deep(p->lchild),Deep(p->rchild))+1;
head->d = max(Deep(head->lchild),Deep(head->rchild))+1;
return p;
}
struct node *LR(struct node *head)
{
head->lchild = RR(head->lchild);//先做左旋
return LL(head);//在做右旋
}
struct node *RL(struct node *head)
{
head->rchild = LL(head->rchild);//先做右旋
return RR(head);//在做左旋
}
struct node *Creat(struct node *head,int x)
{
if (head == NULL)
{
head = (struct node *)malloc (sizeof (struct node));
head->data = x;
head->d = 0;
head->rchild = head->lchild = NULL;
}
else if (head->data > x)
{
head->lchild = Creat(head->lchild,x);
if (Deep(head->lchild)-Deep(head->rchild)>1)
{
if (head->lchild->data > x)
head = LL(head);
else
head = LR(head);
}
}
else if (head->data < x)
{
head->rchild = Creat(head->rchild,x);
if (Deep(head->rchild)-Deep(head->lchild)>1)
{
if (head->rchild->data > x)
head = RL(head);
else
head = RR(head);
}
}
head->d = max(Deep(head->lchild),Deep(head->rchild))+1;
return head;
}
int main()
{
int n,m;
scanf ("%d",&n);
struct node *head = NULL;
for (int i=0; i<n; i++)
{
scanf ("%d",&m);
head = Creat(head,m);
}
printf ("%d\n",head->data);
return 0;
}