AVL树的C语言实现

AVL树是带有平衡条件的二叉查找树。之所以需要平衡是因为有的二叉查找树很糟糕,像这样

 

上一篇文章中二叉查找树的删除总是用右子树的左子树叶子来代替删除节点操作容易造成左子树比右子树深很多的情况(每次删除的都是右子树的节点)。一颗AVL树是其每个节点的左子树和右子树高度最多差1的二叉查找树。空树的高度定义为-1.因为添加了平衡条件,所以我们在编写AVL树的基本操作时需要加入是否平衡的判断,以及解决方法。

解决平衡的方法我们称之为旋转,旋转又分为以下几种情况:

 

这是单旋转,而且它的不平衡是在左边,所以是左单旋转,相应的还有右单旋转。单旋转就相当于将不平衡的三个节点中间的节点给拎了起来,而不平衡的节点就顺着重力掉了下去。单旋转只适用于不平衡节点只有左或者右子树的情况下。这种单旋转可以降低子树的深度。还有一种情况是这样

 

因为子树Y的深度过深,单旋转没有降低它的深度,旋转完之后还是产生了不平衡。这种情况就要用双旋转:双旋转是将最深的子树提升到根的高度从而来降低它的深度。而它的左右子树将相应的分给之前的节点。像这样:

 


这是左双旋转,右双旋转位置和它相反即可。因为我们已经确定某棵树在执行单一操作之前就已经是AVL树,所以造成不平衡的情况只有这些。

下面给出具体的实现代码:

 头文件:

#ifndef __AVLTREE_H
#define __AVLTREE_H

#define Max(a,b) (((a)>(b))?(a):(b)) 

struct AVLNode;
typedef struct AVLNode *AVLposition;
typedef struct AVLNode *AVLTree;
typedef int ElementType;

struct AVLNode
{
	ElementType element;
	AVLTree left;
	AVLTree right;
	int height;

};


AVLTree AVLmakeEmpty(AVLTree T);
AVLposition AVLfind(ElementType x,AVLTree T);
AVLposition AVLfindMax(AVLTree T);
AVLposition AVLfindMin(AVLTree T);
AVLTree AVLinsert(ElementType x,AVLTree T);
AVLTree AVLDelete(ElementType x,AVLTree T);
AVLTree rotate(AVLTree T);

static revolve(ElementType x,AVLTree T);	//旋转
static int getHeight(AVLposition p);
static AVLposition singleRotateWithLeft(AVLposition k2);
static AVLposition singleRotateWithRight(AVLposition k2);
static AVLposition doubleRotateWithLeft(AVLposition k3);
static AVLposition doubleRotateWithRight(AVLposition k3);
void AVLpreOrderTraverse(AVLTree Tree);	//先序遍历 根左右

#endif

实现文件:

#include "AVLTree.h"
#include <stdlib.h>
#include <stdio.h>

AVLTree AVLmakeEmpty(AVLTree T)
{
	if(T)
	{
		AVLmakeEmpty(T->left);
		AVLmakeEmpty(T->right);
		free(T);
	}
	return NULL;
}

AVLposition AVLfind(ElementType x,AVLTree T)
{
	if(!T)
	{
		printf("cant search in a empty tree\n");
		exit(-1);
	}
	else
	{
		if(x<T->element)
			AVLfind(x,T->left);
		else if(x>T->element)
			AVLfind(x,T->right);
		else
			return T;				
	}
}

AVLposition AVLfindMax(AVLTree T)
{
	if(!T)
		return NULL;
	else if(T->right)
		AVLfindMax(T->right);
	else if(T->right==NULL)
		return T;
		
}

AVLposition AVLfindMin(AVLTree T)
{
	if(!T)
		return NULL;
	else if(T->left)
		AVLfindMin(T->left);
	else if(T->left==NULL)
		return T;
}

AVLTree AVLinsert(ElementType x,AVLTree T)
{
	if(T==NULL)
	{
		T=(struct AVLNode*)malloc(sizeof(struct AVLNode));
		if(T==NULL)
		{
			printf("out of space\n");
			exit(-1);
		}
		else
		{
			T->left=T->right=NULL;
			T->element=x;
			T->height=0;
		}
	}
	else if(x<T->element)
	{
		T->left=AVLinsert(x,T->left);
		if(getHeight(T->left)-getHeight(T->right)==2)
		{
			if(x<T->left->element)
				T=singleRotateWithLeft(T);
			else
				T=doubleRotateWithLeft(T);
		}
	}
	else if(x>T->element)
	{
		T->right=AVLinsert(x,T->right);
		if(getHeight(T->right)-getHeight(T->left)==2)
		{
			if(x>T->right->element)
				T=singleRotateWithRight(T);
			else
				T=doubleRotateWithRight(T);
		}
	}

	T->height=Max(getHeight(T->left),getHeight(T->right))+1;
	
	return T;
}

AVLTree AVLDelete(ElementType x,AVLTree T)
{
	AVLTree temp;
	if(T==NULL)
		return NULL;
	if(x==T->element)
	{
		if(T->right==NULL)
		{
			temp=T;
			T=T->left;
			free(temp);
		}
		else
		{
			temp=T->right;
			while(temp->left!=NULL)
				temp=temp->left;
		
		T->element=temp->element;
		T->right=AVLDelete(temp->element,T->right);
		T->height=Max(getHeight(T->left),getHeight(T->right))+1;
		}
		return T;
	}
	else if(x<T->element)
		T->left=AVLDelete(x,T->left);
	else
		T->right=AVLDelete(x,T->right);
	T->height=Max(getHeight(T->left),getHeight(T->right))+1;
	if(T->left!=NULL)
		T->left=rotate(T->left);
	if(T->right!=NULL)
		T->right=rotate(T->right);
	T=rotate(T);

	return T;	
}

AVLTree rotate(AVLTree T)
{
 if(getHeight(T->left) - getHeight(T->right) == 2){
        if(getHeight(T->left->left) >= getHeight(T->left->right)){
            T = singleRotateWithLeft(T);
        }
        else{
            T = doubleRotateWithLeft(T);
        }
    }
    if(getHeight(T->right) - getHeight(T->left) ==2){
        if(getHeight(T->right->right) >= getHeight(T->right->left)){
            T = singleRotateWithRight(T);
        }
        else{
            T = doubleRotateWithRight(T);
        }
    }
    return T;
}

static int getHeight(AVLposition p)
{
	if(p==NULL)
		return -1;
	else
		return p->height;
}

/*this function can be called only if k2 has a left
child,perform a rotate between a node(k2) and its left child
update heights,then return new root
*/
static AVLposition singleRotateWithLeft(AVLposition k2)
{
	AVLposition k1;
	k1=k2->left;
	k2->left=k1->right;
	k1->right=k2;

	/*更新高度为什么是这样*/
	k2->height=Max(getHeight(k2->left),getHeight(k2->right))+1;
	k1->height=Max(getHeight(k1->left),k2->height)+1;
	return k1;
}
static AVLposition singleRotateWithRight(AVLposition k2)
{
	AVLposition k1;
	k1=k2->right;
	k2->right=k1->left;
	k1->left=k2;

	/*更新高度为什么是这样*/
	k2->height=Max(getHeight(k2->left),getHeight(k2->right))+1;
	k1->height=Max(getHeight(k1->left),k2->height)+1;
	return k1;
}

/*this function can be called only if k3 has a left,child and k3's left child has
a right child,Do the left-right double rotation,update heights,then return new 
root*/

static AVLposition doubleRotateWithLeft(AVLposition k3)
{
	k3->left=singleRotateWithRight(k3->left);

	return singleRotateWithLeft(k3);
}
static AVLposition doubleRotateWithRight(AVLposition k3)
{
	k3->right=singleRotateWithLeft(k3->right);

	return singleRotateWithRight(k3);
}

static revolve(ElementType x,AVLTree T)
{
	if(getHeight(T->left)-getHeight(T->right)==2)
	{
		if(x>T->left->element)
			T=singleRotateWithLeft(T);
		else
			T=doubleRotateWithLeft(T);
	}
	if(getHeight(T->right)-getHeight(T->left)==2)
	{
		if(x>T->left->element)
			T=singleRotateWithRight(T);
		else
			T=doubleRotateWithRight(T);

	}
}
void AVLpreOrderTraverse(AVLTree Tree)
{
	if(Tree)
	{
		printf(" %d ",Tree->element);
		AVLpreOrderTraverse(Tree->left);
		AVLpreOrderTraverse(Tree->right);
	}

}	

测试文件:

#include "AVLTree.h"
#include <stdio.h>

int main()
{

	/*******************AVL树测试***************************/
	
	//AVLposition pos;
	AVLTree root=NULL;
	root=AVLinsert(3,root);
	root=AVLinsert(2,root);
	root=AVLinsert(1,root);	//旋转完后,根变了
	root=AVLinsert(4,root);
	root=AVLinsert(5,root);
	root=AVLinsert(6,root);
	root=AVLinsert(7,root);
	/*
	root=AVLinsert(16,root);
	root=AVLinsert(15,root);
	root=AVLinsert(14,root);
	root=AVLinsert(13,root);
	root=AVLinsert(12,root);
	root=AVLinsert(11,root);
	root=AVLinsert(9,root);
	root=AVLinsert(8,root);
	*/
	AVLpreOrderTraverse(root);
	root=AVLDelete(4,root);
	/*root=AVLDelete(3,root);
	root=AVLDelete(1,root);*/
	printf("\n");
	AVLpreOrderTraverse(root);
	printf("\n");

	/**********************************************/

	


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值