平衡二叉树的旋转问题
转 http://gengning938.blog.163.com/blog/static/12822538120114164564057/
对于二叉查找树,尽管查找、插入及删除操作的平均运行时间为O(logn),但是它们的最差运行时间都是O(n),原因在于对树的形状没有限制。
平衡二叉树又称为AVL树,它或者是一棵空树,或者是有下列性质的二叉树:它的左子树和右子树都是平衡二叉树,且左右子树的深度之差的绝对值不超过1。二叉树的的平衡因子BF为:该结点的左子树的深度减去它的右子树的深度,则平衡二叉树的所有结点的平衡因子为只可能是:-1、0和1
一棵好的平衡二叉树的特征:
(1)保证有n个结点的树的高度为O(logn)
(2)容易维护,也就是说,在做数据项的插入或删除操作时,为平衡树所做的一些辅助操作时间开销为O(1)
一、平衡二叉树的构造
在一棵二叉查找树中插入结点后,调整其为平衡二叉树。若向平衡二叉树中插入一个新结点后破坏了平衡二叉树的平衡性。首先要找出插入新结点后失去平衡的最小子树根结点的指针。然后再调整这个子树中有关结点之间的链接关系,使之成为新的平衡子树。当失去平衡的最小子树被调整为平衡子树后,原有其他所有不平衡子树无需调整,整个二叉排序树就又成为一棵平衡二叉树
1.调整方法
(1)插入点位置必须满足二叉查找树的性质,即任意一棵子树的左结点都小于根结点,右结点大于根结点
(2)找出插入结点后不平衡的最小二叉树进行调整,如果是整个树不平衡,才进行整个树的调整。
2.调整方式
(1)LL型
LL型:插入位置为左子树的左结点,进行向右旋转
![平衡二叉树 - 枫叶 - 枫叶](http://img115.ph.126.net/SqbMtCG391VVP3FXrnR_bg==/2010012808692889579.gif)
由于在A的左孩子B的左子树上插入结点F,使A的平衡因子由1变为2,成为不平衡的最小二叉树根结点。此时A结点顺时针右旋转,旋转过程中遵循“旋转优先”的规则,A结点替换D结点成为B结点的右子树,D结点成为A结点的左孩子。
(2)RR型
RR型:插入位置为右子树的右孩子,进行向左旋转
![平衡二叉树 - 枫叶 - 枫叶](http://img609.ph.126.net/hS3sNmPFqzhsRN5AVZQECQ==/1891511843497439874.gif)
由于在A的右子树C的右子树插入了结点F,A的平衡因子由-1变为-2,成为不平衡的最小二叉树根结点。此时,A结点逆时针左旋转,遵循“旋转优先”的规则,A结点替换D结点成为C的左子树,D结点成为A的右子树。
(3)LR型
LR型:插入位置为左子树的右孩子,要进行两次旋转,先左旋转,再右旋转;第一次最小不平衡子树的根结点先不动,调整插入结点所在的子树,第二次再调整最小不平衡子树。
![平衡二叉树 - 枫叶 - 枫叶](http://img.ph.126.net/Q-WMPZXSJAhOiyI9OuqXPQ==/3372914645925000410.gif)
(4)RL型
RL型:插入位置为右子树的左孩子,进行两次调整,先右旋转再左旋转;处理情况与LR类似。
![平衡二叉树 - 枫叶 - 枫叶](http://img74.ph.126.net/QCtqXv7LRpZ78m-nrdy6zA==/1852386821735086852.gif)
#include<iostream>
#include<fstream>
#include<malloc.h>
using namespace std;
#define LH 1//左高
#define EH 0//等高
#define RH -1//右高
struct TreeNode
{
int m_nValue;
int BF;//平衡因子
TreeNode *lchild;
TreeNode *rchild;
};
#include<fstream>
#include<malloc.h>
using namespace std;
#define LH 1//左高
#define EH 0//等高
#define RH -1//右高
struct TreeNode
{
int m_nValue;
int BF;//平衡因子
TreeNode *lchild;
TreeNode *rchild;
};
class AVLTree
{
public:
AVLTree(){};
~AVLTree(){};
void CreateTree(TreeNode *&root,int data);//创建AVL
void PreTraver(TreeNode *root);//先序遍历
void RR_Rotate(TreeNode *&r);//右旋转处理
int GetHeight(TreeNode *root);//获得树的高度
int GetBF(TreeNode *root);//获得树的平衡因子
void LL_Rotate(TreeNode *&r);//左旋转处理
void RL_Rotate(TreeNode *&r);//双旋处理,先右旋转,再左旋转
void LR_Rotate(TreeNode *&r);//双旋处理,先左旋转,再右旋转
void LeftBalance(TreeNode *&T);//左平衡处理
void RightBalance(TreeNode *&T);//右平衡处理
{
public:
AVLTree(){};
~AVLTree(){};
void CreateTree(TreeNode *&root,int data);//创建AVL
void PreTraver(TreeNode *root);//先序遍历
void RR_Rotate(TreeNode *&r);//右旋转处理
int GetHeight(TreeNode *root);//获得树的高度
int GetBF(TreeNode *root);//获得树的平衡因子
void LL_Rotate(TreeNode *&r);//左旋转处理
void RL_Rotate(TreeNode *&r);//双旋处理,先右旋转,再左旋转
void LR_Rotate(TreeNode *&r);//双旋处理,先左旋转,再右旋转
void LeftBalance(TreeNode *&T);//左平衡处理
void RightBalance(TreeNode *&T);//右平衡处理
};
int Max(int a,int b)
{
if(a>b)return a;
else return b;
}
int AVLTree::GetHeight(TreeNode *root)
{
int len;
if(root==NULL)len=0;
else
{
len=Max(GetHeight(root->lchild),GetHeight(root->rchild))+1;
}
return len;
}
int AVLTree::GetBF(TreeNode *root)
{
int bf;//平衡因子
bf=GetHeight(root->lchild)-GetHeight(root->rchild);
return bf;
}
void AVLTree::CreateTree(TreeNode *&root,int data)
{
if(root==NULL)
{
root=(TreeNode*)malloc(sizeof(TreeNode));
root->m_nValue=data;
root->lchild=NULL;
root->rchild=NULL;
// root->BF=GetBF(root);
root->BF=EH;//初始叶子结点的平衡因子为等高
}
else if(data<root->m_nValue)
{
CreateTree(root->lchild,data);
switch(root->BF)//检查root的平衡度
{
case LH://原来树root的左子树比右子树高,现在左子树更高
LeftBalance(root);//对树进行左平衡处理
break;
case EH://原来树root的左右子树等高,现在左子树高
root->BF=LH;//root的平衡因子由0变为1
break;
case RH://原来树root的右子树比左子树高,现在左右子树等高
root->BF=EH;
}
}
else if(data>root->m_nValue)
{
CreateTree(root->rchild,data);
switch(root->BF)
{
case LH:root->BF=EH;//原来树root的左子树比右子树高,现在root的左右子树等高
break;
case EH://原来树root的左右子树等高,现在root的右子树更高
root->BF=RH;
break;
case RH://原来右子树比左子树高,现在root右子树高
RightBalance(root);//对树root作右平衡处理
}
}
}
void AVLTree::PreTraver(TreeNode *root)
{
if(root)
{ cout.width(3);
cout<<root->m_nValue;
}
if(root->lchild)
PreTraver(root->lchild);
if(root->rchild)
PreTraver(root->rchild);
}
void AVLTree::LL_Rotate(TreeNode *&r)//插入位置为右子树右孩子,要进行左旋转
{
TreeNode *p;
p=r->rchild;//p指向r的右孩子结点
r->rchild=p->lchild;//r结点左旋转成为p的左子树,p原来的左子树成为r的右子树
p->lchild=r;//r成为p的左孩子
r=p;
}
void AVLTree::RR_Rotate(TreeNode *&r)//插入位置为左子树左孩子,进行右旋转
{
TreeNode *p;
p=r->lchild;
r->lchild=p->rchild;
p->rchild=r;
r=p;
}
void AVLTree::RL_Rotate(TreeNode *&r)//插入位置为右子树左孩子,先进行右旋转,再进行左旋转
{
TreeNode *p;
p=r->rchild;
RR_Rotate(p);//最小失衡树的根结点的右子树根结点进行右旋转
r->rchild=p;//更新最小失衡树根结点的右孩子
LL_Rotate(r);//最小失衡树的根结点进行左旋转
}
void AVLTree::LR_Rotate(TreeNode *&r)//插入位置为左子树右孩子,先进行左旋转,再进行右旋转
{
TreeNode *p;
p=r->lchild;
LL_Rotate(p);//最小失衡树根结点的左子树根结点进行左旋转
r->lchild=p;//更新最小失衡树根结点的左孩子
RR_Rotate(r);//最小失衡树根结点进行右旋转
}
void AVLTree::LeftBalance(TreeNode *&T)//左平衡处理
{//初始条件:原来平衡的二叉排序树T的左子树比右子树高(T->bf=1)
// 又在左子树中插入了结点,并导致左子树更高,破坏了树T的平衡性
//操作结果:对不平衡的树T作左平衡旋转处理,使树T的重心右移实现 新的平衡
TreeNode *lc,*rd;
lc=T->lchild;//lc指向T的左孩子结点
switch(lc->BF)//检查T左子树的平衡因子
{
case LH://新结点插入在T的左孩子的左子树上,导致左子树的平衡因子为左高,进行右旋转处理
T->BF=lc->BF=EH;//旋转后,原根结点和左孩子结点平衡因子都 为0
RR_Rotate(T);//右旋转处理
break;
case RH://新结点插入在T的左孩子的右子树上,导致左子树的平衡因子为右高,进行LR处理
rd=lc->rchild;
switch(rd->BF)
{
case LH://新结点插入在T的左孩子的右子树的左子树上
T->BF=RH;//旋转后,原根结点的平衡因子为右高
lc->BF=EH;//旋转后,原根结点的左孩子结点平衡因子为等高
break;
case EH://新结点插入到T的左孩子的右孩子(叶子)
T->BF=lc->BF=EH;//旋转后,原根和左孩子结点的平衡因子都为等高
break;
case RH://新结点插入在T的左孩子的右子树的右子树上
T->BF=EH;//旋转后,原根结点的平衡因子为等高
lc->BF=LH;//旋转后,原根结点的左孩子结点平衡因子为左高
}
rd->BF=EH;//旋转后的新结点的平衡因子为等高
int Max(int a,int b)
{
if(a>b)return a;
else return b;
}
int AVLTree::GetHeight(TreeNode *root)
{
int len;
if(root==NULL)len=0;
else
{
len=Max(GetHeight(root->lchild),GetHeight(root->rchild))+1;
}
return len;
}
int AVLTree::GetBF(TreeNode *root)
{
int bf;//平衡因子
bf=GetHeight(root->lchild)-GetHeight(root->rchild);
return bf;
}
void AVLTree::CreateTree(TreeNode *&root,int data)
{
if(root==NULL)
{
root=(TreeNode*)malloc(sizeof(TreeNode));
root->m_nValue=data;
root->lchild=NULL;
root->rchild=NULL;
// root->BF=GetBF(root);
root->BF=EH;//初始叶子结点的平衡因子为等高
}
else if(data<root->m_nValue)
{
CreateTree(root->lchild,data);
switch(root->BF)//检查root的平衡度
{
case LH://原来树root的左子树比右子树高,现在左子树更高
LeftBalance(root);//对树进行左平衡处理
break;
case EH://原来树root的左右子树等高,现在左子树高
root->BF=LH;//root的平衡因子由0变为1
break;
case RH://原来树root的右子树比左子树高,现在左右子树等高
root->BF=EH;
}
}
else if(data>root->m_nValue)
{
CreateTree(root->rchild,data);
switch(root->BF)
{
case LH:root->BF=EH;//原来树root的左子树比右子树高,现在root的左右子树等高
break;
case EH://原来树root的左右子树等高,现在root的右子树更高
root->BF=RH;
break;
case RH://原来右子树比左子树高,现在root右子树高
RightBalance(root);//对树root作右平衡处理
}
}
}
void AVLTree::PreTraver(TreeNode *root)
{
if(root)
{ cout.width(3);
cout<<root->m_nValue;
}
if(root->lchild)
PreTraver(root->lchild);
if(root->rchild)
PreTraver(root->rchild);
}
void AVLTree::LL_Rotate(TreeNode *&r)//插入位置为右子树右孩子,要进行左旋转
{
TreeNode *p;
p=r->rchild;//p指向r的右孩子结点
r->rchild=p->lchild;//r结点左旋转成为p的左子树,p原来的左子树成为r的右子树
p->lchild=r;//r成为p的左孩子
r=p;
}
void AVLTree::RR_Rotate(TreeNode *&r)//插入位置为左子树左孩子,进行右旋转
{
TreeNode *p;
p=r->lchild;
r->lchild=p->rchild;
p->rchild=r;
r=p;
}
void AVLTree::RL_Rotate(TreeNode *&r)//插入位置为右子树左孩子,先进行右旋转,再进行左旋转
{
TreeNode *p;
p=r->rchild;
RR_Rotate(p);//最小失衡树的根结点的右子树根结点进行右旋转
r->rchild=p;//更新最小失衡树根结点的右孩子
LL_Rotate(r);//最小失衡树的根结点进行左旋转
}
void AVLTree::LR_Rotate(TreeNode *&r)//插入位置为左子树右孩子,先进行左旋转,再进行右旋转
{
TreeNode *p;
p=r->lchild;
LL_Rotate(p);//最小失衡树根结点的左子树根结点进行左旋转
r->lchild=p;//更新最小失衡树根结点的左孩子
RR_Rotate(r);//最小失衡树根结点进行右旋转
}
void AVLTree::LeftBalance(TreeNode *&T)//左平衡处理
{//初始条件:原来平衡的二叉排序树T的左子树比右子树高(T->bf=1)
// 又在左子树中插入了结点,并导致左子树更高,破坏了树T的平衡性
//操作结果:对不平衡的树T作左平衡旋转处理,使树T的重心右移实现 新的平衡
TreeNode *lc,*rd;
lc=T->lchild;//lc指向T的左孩子结点
switch(lc->BF)//检查T左子树的平衡因子
{
case LH://新结点插入在T的左孩子的左子树上,导致左子树的平衡因子为左高,进行右旋转处理
T->BF=lc->BF=EH;//旋转后,原根结点和左孩子结点平衡因子都 为0
RR_Rotate(T);//右旋转处理
break;
case RH://新结点插入在T的左孩子的右子树上,导致左子树的平衡因子为右高,进行LR处理
rd=lc->rchild;
switch(rd->BF)
{
case LH://新结点插入在T的左孩子的右子树的左子树上
T->BF=RH;//旋转后,原根结点的平衡因子为右高
lc->BF=EH;//旋转后,原根结点的左孩子结点平衡因子为等高
break;
case EH://新结点插入到T的左孩子的右孩子(叶子)
T->BF=lc->BF=EH;//旋转后,原根和左孩子结点的平衡因子都为等高
break;
case RH://新结点插入在T的左孩子的右子树的右子树上
T->BF=EH;//旋转后,原根结点的平衡因子为等高
lc->BF=LH;//旋转后,原根结点的左孩子结点平衡因子为左高
}
rd->BF=EH;//旋转后的新结点的平衡因子为等高
//双旋转处理
LL_Rotate(T->lchild);//对T的左子树左旋转处理
RR_Rotate(T);//对T作右旋转处理
LL_Rotate(T->lchild);//对T的左子树左旋转处理
RR_Rotate(T);//对T作右旋转处理
}
}
}
void AVLTree::RightBalance(TreeNode *&T)//右平衡处理
{//初始条件:原来平衡二叉排序树T的右子树比左子树高,又在右子树中插入结点,导致右子树更高
//操作结果:对不平衡的树T作右平衡旋转处理
TreeNode *rc,*ld;
rc=T->rchild;
switch(rc->BF)
{
case RH://新结点插入在T的右孩子的右子树上,导致右子平衡因子为右高,进行左旋转处理
T->BF=rc->BF=EH;//旋转后,原根结点和右孩子结点的平衡因子均为0
LL_Rotate(T);
break;
case LH://新结点插入在T的右孩子的左子树上,导致右子树的平衡因子为左高,进行双旋处理
ld=rc->lchild;
switch(ld->BF)
{
case RH://新结点插入在T的右孩子的左子树的右子树上
T->BF=LH;//旋转后,原根结点的平衡因子为左高
rc->BF=EH;//旋转后,原根结点的右孩子结点平衡因子为等高
break;
case EH://新结点插入到T的右孩子的左孩子(叶子)
T->BF=rc->BF=EH;//旋转后,原根和右孩子结点的平衡因子等高
break;
case LH://新结点插入到T的右孩子的左子树的左子树
T->BF=EH;//旋转后,原根结点的平衡因子等高
rc->BF=RH;//旋转后,原根结点的右孩子结点的平衡因子为右高
}
ld->BF=EH;//旋转后的新根结点的平衡因子为等高
{//初始条件:原来平衡二叉排序树T的右子树比左子树高,又在右子树中插入结点,导致右子树更高
//操作结果:对不平衡的树T作右平衡旋转处理
TreeNode *rc,*ld;
rc=T->rchild;
switch(rc->BF)
{
case RH://新结点插入在T的右孩子的右子树上,导致右子平衡因子为右高,进行左旋转处理
T->BF=rc->BF=EH;//旋转后,原根结点和右孩子结点的平衡因子均为0
LL_Rotate(T);
break;
case LH://新结点插入在T的右孩子的左子树上,导致右子树的平衡因子为左高,进行双旋处理
ld=rc->lchild;
switch(ld->BF)
{
case RH://新结点插入在T的右孩子的左子树的右子树上
T->BF=LH;//旋转后,原根结点的平衡因子为左高
rc->BF=EH;//旋转后,原根结点的右孩子结点平衡因子为等高
break;
case EH://新结点插入到T的右孩子的左孩子(叶子)
T->BF=rc->BF=EH;//旋转后,原根和右孩子结点的平衡因子等高
break;
case LH://新结点插入到T的右孩子的左子树的左子树
T->BF=EH;//旋转后,原根结点的平衡因子等高
rc->BF=RH;//旋转后,原根结点的右孩子结点的平衡因子为右高
}
ld->BF=EH;//旋转后的新根结点的平衡因子为等高
//双旋转处理
RR_Rotate(T->rchild);//对T的右子树作右旋转处理
LL_Rotate(T);//对T作左旋转处理
}
}
RR_Rotate(T->rchild);//对T的右子树作右旋转处理
LL_Rotate(T);//对T作左旋转处理
}
}
int main()
{
const char *file="data.txt";
AVLTree AVL=AVLTree();
TreeNode *root=NULL;
int data;
ifstream fin;
fin.open(file);
if(fin.is_open())
{
while(1)
{
fin>>data;
if(data==-1)break;
AVL.CreateTree(root,data);
}
}
AVL.PreTraver(root);
cout<<endl<<endl;
fin.close();
return 1;
}