平衡二叉树详解 图文教程(AVL) 数据结构:(概念篇+完整代码 001)

 

 

 

 

 

一、理解

这个是一个微信公众号写的,我觉得写的通俗易懂。

 

 

1、若它的左子树不为空,则左子树上所有的节点值都小于它的根节点值。

 

2、若它的右子树不为空,则右子树上所有的节点值均大于它的根节点值。

 

3、它的左右子树也分别可以充当为二叉查找树。

 

例如:

 

 

 

 

例如,我现在想要查找数值为14的节点。由于二叉查找树的特性,我们可以很快着找到它,其过程如下:

 

1、和根节点9比较

 

 

2、由于 14 > 9,所以14只可能存在于9的右子树中,因此查看右孩子13

 

 

3、由于 14 > 13,所以继续查看13的右孩子15

 

 

4、由于 14 < 15,所以14只可能存在于15的左孩子中,因此查找15的左孩子14

 

 

5、这时候发现14正是自己查找的值,于是查找结束。

 

这种查找二叉树的查找正是二分查找的思想,可以很快着找到目的节点,查找所需的最大次数等同于二叉查找树的高度。

 

在插入的时候也是一样,通过一层一层的比较,最后找到适合自己的位置。

 

 

 

 

初始的二叉查找树只有三个节点:

 

 

然后我们按照顺序陆续插入节点 4,3,2,1,0。插入之后的结构如下:

 

 

如何是线性的,n为节点个数,时间复杂度为O(n),如果用AVL的话,可以达到O(logn)

 

 

 

 

这是一种比查找二叉树还特别的树哦,这种树就可以帮助我们解决二叉查找树刚才的那种所有节点都倾向一边的缺点的。具有如下特性:

 

  1. 具有二叉查找树的全部特性。

  2. 每个节点的左子树和右子树的高度差至多等于1。

 

例如:图一就是一颗AVL树了,而图二则不是(节点右边标的是这个节点的高度)。

 

 

 

对于图二,因为节点9的左孩子高度为2,而右孩子高度为0。他们之间的差值超过1了。

 

这种树就可以保证不会出现大量节点偏向于一边的情况了。

 

听起来这种树还不错,可以对于图1,如果我们要插入一个节点3,按照查找二叉树的特性,我们只能把3作为节点4的左子树插进去,可是插进去之后,又会破坏了AVL树的特性,那我们那该怎么弄?

 

右旋

 

我们在进行节点插入的时候,可能会出现节点都倾向于左边的情况,例如:

 

我们把这种倾向于左边的情况称之为 左-左型(LL)。这个时候,我们就可以对节点9进行右旋操作,使它恢复平衡。

如何理解这个右旋呢?

我是这样理解的:

相对于9来说,让5绕着9顺时针向右旋转,成为新的root

 

 

即:顺时针旋转两个节点,使得父节点被自己的左孩子取代,而自己成为自己的右孩子

 

再举个例子:

 

节点4和9高度相差大于1。由于是左孩子的高度较高,此时是左-左(LL)型,进行右旋。

 

 

这里要注意,节点4的右孩子成为了节点6的左孩子了

 

我找了个动图,尽量这个动图和上面例子的节点不一样。

 

左旋

 

左旋和右旋一样,就是用来解决当大部分节点都偏向右边的时候,通过左旋来还原。例如:

 

 

我们把这种倾向于右边的情况称之为 右-右(RR)型

 

我也找了一张动图。

 

 

例子讲解

 

 

初始状态如下:

 

 

然后我们主键插入如下数值:1,4,5,6,7,10,9,8

 

插入 1

 

 

左-左型,需要右旋调整。

 

 

插入4

 

 

继续插入 5

 

右-右型,需要左旋转调整。

 

 

继续插入6

 

右-右型,需要进行左旋

 

继续插入7

 

右-右型,需要进行左旋

 

 

继续插入10

 

 

继续插入9

 

 

出现了这种情况怎么办呢?对于这种  右-左型 的情况,单单一次左旋或右旋是不行的,下面我们先说说如何处理这种情况。

 

 

这种我们就把它称之为 右-左 型吧。处理的方法是先对节点10进行右旋把它变成右-右型。

 

然后在进行左旋。

 

 

所以对于这种 右-左型的,我们需要进行一次右旋再左旋

 

同理,也存在 左-右型的,例如:

 

 

对于左-右型的情况和刚才的 右-左型相反,我们需要对它进行一次左旋,再右旋。

 

 

回到刚才那道题

 

 

对它进行右旋再左旋。

 

 

到此,我们的插入就结束了。

 

 

 

总结一下

 

在插入的过程中,会出现一下四种情况破坏AVL树的特性,我们可以采取如下相应的旋转。

 

1、左-左(LL)型:做右旋。

2、右-右(RR)型:做左旋转。

3、左-右(LR)型:先做左旋,后做右旋。

4、右-左(RL)型:先做右旋,再做左旋。

 

不知道大家发现规律没,这个规则还是挺好记。

 

 

补充:

平衡因子:左右子树高度差就是当前节点的平衡因子。即:拿着左子树的高度-右子树的高度。

AVL中平衡因子的取值范围【-2,2】

 

 

 

 

 

二、代码实现

1.存储结构

比二叉搜索树多了一个height,用来记录当前节点的高度。

struct node{
	int data;
	int height;
	node* lchild;
	node* rchild;
};

2.新生成节点

新生成节点的高度都是1,后面会更新。

node* newnode(int v){
	node* t=new node;
	t->data=v;
	t->height=1;
	t->lchild=t->rchild=NULL;
	return t; 
}

3.获得高度值以及平衡因子

int getheight(node* root){
	if(root==NULL){
		return 0;
	}
	return root->height;
}
int getBalanceFactor(node* root){
	return getheight(root->lchild)-getheight(root->rchild);
}

 

4.更新高度值

每次插入新节点后,都需要更新高度值。

void update_height(node* &root){
	root->height=max(getheight(root->lchild),getheight(root->rchild))+1;
}

5.搜索

void search(node* root,int v){
	if(root==NULL){
		return ;
	}
	if(root->data==v){
		cout<<"search success!"<<endl;
		return ;
	}
	else
	if(root->data>v){
		search(root->lchild,v);
	}
	else
	if(root->data<v){
		search(root->rchild,v);
	}
}

6.旋转

6.1左旋

void L(node* &root){     //左旋 
	node* t=root->rchild;    
	root->rchild=t->lchild;
	t->lchild=root;
	update_height(root);    // 更新节点的高度 
	update_height(t);
	root=t;
}

6.2右旋

void R(node* &root){      // 右旋 
	node* t=root->lchild;
	root->lchild=t->rchild;
	t->rchild=root;
	update_height(root);
	update_height(t);
	root=t;
}

7.插入

注:BF表示平衡因子

树形判定条件调整方法
LLBF(root)==2&&BF(root->lchild)==1对root进行右旋
LRBF(root)==2&&BF(root->lchild)==-1对root的左子树进行左旋,再对root进行右旋
RRBF(root)==-2&&BF(root->rchild)==-1对root进行左旋
RLBF(root)==-2&&BF  (root->lchild)==1对root的右子树进行右旋,再对root进行左旋
void insert(node* &root,int v){
	if(root==NULL){
		root=newnode(v);
		return ;
	}
	if(root->data>v){
		insert(root->lchild,v);
		update_height(root);
		if(getBalanceFactor(root)==2){
			if(getBalanceFactor(root->lchild)==1){
				R(root);
			}
			else
			if(getBalanceFactor(root->lchild)==-1){
				L(root->lchild);
				R(root);
			}
		}
	}	
	else
	{
		insert(root->rchild,v);
		update_height(root);
		if(getBalanceFactor(root)==-2){
			if(getBalanceFactor(root->rchild)==1){
				R(root->lchild);
				L(root);
			}
			else
			if(getBalanceFactor(root->rchild)==-1){
				L(root);
			}
		}
	}
}    

切记:每插入一个节点,都要更新高度值,判断是否符合AVL的定义,再进行旋转。

8.建树

node* create(int data[],int len){
	node* root=NULL;
	for(int i=0;i<len;i++){
		insert(root,data[i]);
	}
	return root;
} 

三、完整的代码

 

#include<iostream>
#include<algorithm>
using namespace std;
#define NULL 0
#define range 102
struct node{
	int data;
	int height;
	node* lchild;
	node* rchild;
};
int data[range];
int n;
node* Root=NULL; 
node* newnode(int v){
	node* t=new node;
	t->data=v;
	t->height=1;
	t->lchild=t->rchild=NULL;
	return t; 
}
int getheight(node* root){
	if(root==NULL){
		return 0;
	}
	return root->height;
}
int getBalanceFactor(node* root){
	return getheight(root->lchild)-getheight(root->rchild);
}
void update_height(node* &root){
	root->height=max(getheight(root->lchild),getheight(root->rchild))+1;
}
void search(node* root,int v){
	if(root==NULL){
		return ;
	}
	if(root->data==v){
		cout<<"search success!"<<endl;
		return ;
	}
	else
	if(root->data>v){
		search(root->lchild,v);
	}
	else
	if(root->data<v){
		search(root->rchild,v);
	}
}
void L(node* &root){     //左旋 
	node* t=root->rchild;    
	root->rchild=t->lchild;
	t->lchild=root;
	update_height(root);    // 更新节点的高度 
	update_height(t);
	root=t;
}
void R(node* &root){      // 右旋 
	node* t=root->lchild;
	root->lchild=t->rchild;
	t->rchild=root;
	update_height(root);
	update_height(t);
	root=t;
}           
void insert(node* &root,int v){
	if(root==NULL){
		root=newnode(v);
		return ;
	}
	if(root->data>v){
		insert(root->lchild,v);
		update_height(root);
		if(getBalanceFactor(root)==2){
			if(getBalanceFactor(root->lchild)==1){
				R(root);
			}
			else
			if(getBalanceFactor(root->lchild)==-1){
				L(root->lchild);
				R(root);
			}
		}
	}	
	else
	{
		insert(root->rchild,v);
		update_height(root);
		if(getBalanceFactor(root)==-2){
			if(getBalanceFactor(root->rchild)==1){
				R(root->lchild);
				L(root);
			}
			else
			if(getBalanceFactor(root->rchild)==-1){
				L(root);
			}
		}
	}
}    
node* create(int data[],int len){
	node* root=NULL;
	for(int i=0;i<len;i++){
		insert(root,data[i]);
	}
	return root;
}      
void preorder(node* root){
	if(root==NULL){
		return ;
	}
	cout<<root->data<<"  平衡因子为:"<<getBalanceFactor(root)<<endl;
	preorder(root->lchild);
	preorder(root->rchild);
} 
void inorder(node* root){
	if(root==NULL){
		return ;
	}
	inorder(root->lchild);
	cout<<root->data<<"  平衡因子为:"<<getBalanceFactor(root)<<endl;
	
	inorder(root->rchild);
}                         
int main(){
	int i,j;
	cout<<"请输入节点个数:";
	cin>>n;
	cout<<"请输入节点值:";
	for(i=0;i<n;i++)
	{
		cin>>data[i];
	}
	Root=create(data,n);
	cout<<"先序遍历:"<<endl;
	preorder(Root);
	cout<<"中序遍历:"<<endl;
	inorder(Root);
	return 0;
}

 

 

 

 

 

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值