平衡二叉树(AVL树)

平衡二叉树:是一颗空树;或者具有以下性质的树:它的左子树和右子树都是平衡二叉树,且左子树和右子树的深度之差的绝对值不超过1。

平衡二叉树的关键在于插入结点时如何保持整棵树的平衡性。

下面是不平衡发生的四种情况:

(1)平衡二叉树某一节点的左孩子的左子树上插入一个新的节点,使得该节点不再平衡。

LL型(左孩子的左子树)
由于在A的左孩子B的左子树上插入结点F,使A的平衡因子由1增至2而失去平衡。故需进行一次顺时针旋转操作。 即将A的左孩子B向右上旋转代替A作为根结点,A向右下旋转成为B的右子树的根结点。而原来B的右子树则变成A的左子树。


/* return pointer to the new subtree root */
AVLNode *single_rotate_ll(AVLNode *node)	 //LL
{
	AVLNode *p = node->lchild;

	node->lchild = p->rchild;
	p->rchild = node;
	
	node->height = max(height(node->lchild), height(node->rchild)) + 1;
	p->height = max(height(p->lchild), height(p->rchild)) + 1;
	//p->height = max(height(p->lchild), node->height) + 1;

	return p;
}

(2) 平衡二叉树某一节点的右孩子的右子树上插入一个新的节点,使得该节点不再平衡。

RR型平衡旋转法(右孩子的右子树)

由于在A的右孩子C 的右子树上插入结点F,使A的平衡因子由-1减至-2而失去平衡。故需进行一次逆时针旋转操作。即将A的右孩子C向左上旋转代替A作为根结点,A向左下旋转成为C的左子树的根结点。而原来C的左子树则变成A的右子树。


AVLNode *single_rotate_rr(AVLNode *node)	 //RR
{
	AVLNode *p = node->rchild;

	node->rchild = p->lchild;
	p->lchild = node;

	node->height = max(height(node->lchild), height(node->rchild)) + 1;
	p->height = max(height(p->lchild), height(p->rchild)) + 1;
	//p->height = max(node->height, height(p->rchild)) + 1;

	return p;
}

(3)平衡二叉树某一节点的左孩子的右子树上插入一个新的节点,使得该节点不再平衡。

LR型平衡旋转法(左孩子的右子树)
由于在A的左孩子B的右子树上插入结点F,使A的平衡因子由1增至2而失去平衡。故需进行两次旋转操作(先逆时针,后顺时针)。即先将A结点的左孩子B的右子树的根结点D向左上旋转提升到B结点的位置,然后再把该D结点向右上旋转提升到A结点的位置。即先使之成为LL型,再按LL型处理。


AVLNode *double_rotate_lr(AVLNode *node)	//LR
{
	node->lchild = single_rotate_rr(node->lchild);
	return single_rotate_ll(node);
}

(4) 平衡二叉树某一节点的右孩子的左子树上插入一个新的节点,使得该节点不再平衡。

RL型平衡旋转法(右孩子的左子树)
由于在A的右孩子C的左子树上插入结点F,使A的平衡因子由-1减至-2而失去平衡。故需进行两次旋转操作(先顺时针,后逆时针),即先将A结点的右孩子C的左子树的根结点D向右上旋转提升到C结点的位置,然后再把该D结点向左上旋转提升到A结点的位置。即先使之成为RR型,再按RR型处理。


AVLNode *double_rotate_rl(AVLNode *node)	//RL
{
	node->rchild = single_rotate_ll(node->rchild);
	return single_rotate_rr(node);
}
AVl树的结构体为:

typedef struct AVLNode 
{
	int data;
	int height;
	struct AVLNode *lchild, *rchild;
}AVLNode;
计算以node结点为根的树的高度:
int height(AVLNode *node)
{
	if (node == NULL)
		return -1;
	else
		return node->height;
}
现在可以写出avl树的插入代码了:

AVLNode *insert(AVLNode *root, int data)
{
	if (root == NULL)
	{
		root = (AVLNode*)malloc(sizeof(struct AVLNode));
		root->data = data;
		root->lchild = root->rchild = NULL;
		root->height = 0;
	}
	else if (data < root->data)
	{
		root->lchild = insert(root->lchild, data);
		if (height(root->lchild) - height(root->rchild) == 2)   //左子树高度-右子树高度
		{
			if (data < root->lchild->data)	//LL
				root = single_rotate_ll(root);
			else
				root = double_rotate_lr(root);     //LR
		}
	}
	else if (data > root->data)
	{
		root->rchild = insert(root->rchild, data);
		if (height(root->rchild) - height(root->lchild) == 2)	 //右子树高度-左子树高度
		{
			if (data < root->rchild->data)	//RL
				root = double_rotate_rl(root);
			else
				root = single_rotate_rr(root);	   //RR
		}
	}
	//else
	//{}  //data is in the tree already, do nothing

	root->height = max(height(root->lchild), height(root->rchild)) + 1;
	return root;
}


中序遍历avl树:

void inorder_traversal_tree(AVLNode *node)
{
	if (node != NULL)
	{
		inorder_traversal_tree(node->lchild);
		printf("%d ", node->data);
		inorder_traversal_tree(node->rchild);
	}
}


后续遍历以释放树的结点:

void destory_tree(AVLNode*node)
{
	if (node != NULL)
	{
		destory_tree(node->lchild);
		destory_tree(node->rchild);
		free(node), node = NULL;
	}
}


测试代码:

int main()
{
	int arr[] = {3, 2, 1, 4, 5, 6, 7, 16, 15, 14, 13, 12, 11, 10, 8, 9};
	AVLNode* T = NULL;
	for (int i = 0; i < sizeof(arr) / sizeof(arr[0]); i++)
		T = insert(T, arr[i]);

	inorder_traversal_tree(T);
	destory_tree(T);
	getchar();
	return 0;
}

参考: AVL树的旋转

平衡二叉树 AVL 的插入节点后旋转方法分析

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值