AVL树（c实现）

最新推荐文章于 2022-05-11 22:58:33 发布

half-beast

最新推荐文章于 2022-05-11 22:58:33 发布

阅读量615

点赞数

分类专栏：数据结构文章标签： AVL树 c 数据结构

本文链接：https://blog.csdn.net/u010049696/article/details/44487265

版权

数据结构专栏收录该内容

6 篇文章 0 订阅

订阅专栏

本文详细解析了AVL树的C语言实现，包括基础的二叉平衡树概念，AVL树的旋转操作，以及如何通过中根遍历来构建AVL树。重点阐述了AVL树在插入节点时的平衡因子调整机制和树高计算过程，通过实例展示了AVL树的插入过程及其平衡状态的变化。

摘要由CSDN通过智能技术生成

最近阅读java的TreeMap源码，发觉是使用红黑树实现内部的数据管理，所以开始学习编写红黑树。然后发现要想理解红黑树还是要从基础的二叉平衡树的开始。于是，最终决定拿AVL树开刀。

网络上有很多AVL树的各种语言实现版本，个人偏爱c的绝对控制性编程，所以选择了c实现。令人庆幸的是，一个非常简易的AVL树c实现被我找到。现在就在其基础上加上个人理解把AVL树实现描述如下：

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

typedef struct Node {
	int data;
	int height;
	struct Node *lchild,*rchild;
} Node;

int Max(int a,int b)
{
	return a>b?a:b;
}
int height(Node* pNode)
{
	if(pNode == NULL)
	{
		return -1;
	}
	return pNode->height;
}
Node* single_rotate_with_left(Node* pNode)
{
	Node* tmp;

	tmp = pNode->lchild;
	pNode->lchild = tmp->rchild;
	tmp->rchild = pNode;
	//结点的位置改变，需更新结点的高度
	//结点的高度根据孩子的计算得到：
	//pNode的左右孩子分别是pNode->lchild,pNode->rchild
	//tmp的孩子:tmp->lchild,pNode
	pNode->height = Max(height(pNode->lchild),height(pNode->rchild)) + 1;
	tmp->height = Max(height(tmp->lchild),pNode->height) + 1;
	return tmp;
}

Node* single_rotate_with_right(Node* pNode)
{
	Node* tmp;
	tmp = pNode->rchild;
	pNode->rchild = tmp->lchild;
	tmp->lchild = pNode;
	//原理同上
	pNode->height = Max(height(pNode->lchild),height(pNode->rchild)) + 1;
	tmp->height = Max(height(tmp->rchild),pNode->height) + 1;
	return tmp;
}

Node* double_l(Node* pNode)
{
	pNode->lchild = single_rotate_with_right(pNode->lchild);

	return single_rotate_with_left(pNode);
}

Node* double_r(Node* pNode)
{
	pNode->rchild = single_rotate_with_left(pNode->rchild);
	return single_rotate_with_right(pNode);
}

Node* insert(int d,Node* pNode)
{
	if(pNode == NULL)
	{//如果删除判断平衡因子的语句和更新树高度的语句，则此insert完全是一个BST的插入操作
	 //所以从此函数，可以看出AVL树脱胎自BST
		pNode = (Node*)malloc(sizeof(Node));
		pNode->data = d;
		pNode->height = 0;
		pNode->lchild = pNode->rchild = NULL;
	} else if (d <  pNode->data) { //插入左子树
		pNode->lchild = insert(d,pNode->lchild);

		if(height(pNode->lchild) - height(pNode->rchild) == 2) //not balence
		{
			if(d < pNode->lchild->data)
			{
				//插入到左子树，做单旋转
				pNode = single_rotate_with_left(pNode);
			} else {
				pNode = double_l(pNode);
			}
		}
	} else if (d > pNode->data) {
		pNode->rchild = insert(d,pNode->rchild);

		if(height(pNode->rchild) - height(pNode->lchild) == 2) {
			if(d > pNode->rchild->data)
			{
				pNode = single_rotate_with_right(pNode);
			} else {
				pNode = double_r(pNode);
			}
		}
	}

	pNode->height = Max(height(pNode->lchild),height(pNode->rchild)) + 1;
	return pNode;
}
void print(Node* root)
{//中根遍历二叉树
	if(root == NULL)
	{
		return ;
	}
	print(root->lchild);
	fprintf(stdout,"%d ",root->data);
	print(root->rchild);
}

int main(void)
{
	int i;
	Node* root = NULL;

	int a[] = {1,2,3,4,5};
	const int len = sizeof(a)/sizeof(int);
	
	for(;i<len;i++)
	{
		root = insert(a[i],root);
	}

	print(root);
	return 0;
}<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"> </span>

以上代码通过测试。

详解依次插入1,2,3,4,5时的建树过程：

1、插入1：

2、插入2：

3、插入3：

1 1 2

\ ==> \ ==> / \

2 2 1 3

4、插入4：

/ \

1 3

5 、插入5：