AVL(二叉平衡树) 的实现

一颗AVL树是其每个节点的左子树与右子树的高度最多差1的二叉查找树。

在插入过程中，利用旋转的办法保持这个性质。

共分四种情形：

1、  树T的左孩子的左子树上新插入节点导致破坏平衡性：

如下图左边所示，因为在子树X中新加入一个节点，导致k2处的平衡性被破坏

通过如右边所示的旋转，可以使得整棵树重新变得平衡。

2、 树T的右孩子的右子树上新插入节点导致破坏平衡性

这种情形跟上面那种情形是对称的。

3、树T的左孩子的右子树上新插入节点导致破坏平衡性

此时照搬情形1的旋转方法已经不能奏效了。

考虑将Y这部分散开来看：

可经由两次旋转解决问题：

可以发现，经过第一次旋转之后，问题已经变成了情形1，可再进行一次单旋转解决：

4、树T的右孩子的左子树上新插入结点，这时与情形3完全对称，可类似解决。

以下是实现部分。

仅实现插入操作，删除一般是惰性的。还有一点假设相同元素不出现在树结构中。

avl.h

#pragma once
using Elemtype = int;
struct AvlNode;
typedef struct AvlNode *Position;
typedef struct AvlNode *AvlTree;

//AvlTree MakeEmpty(AvlTree T);
//Position Find(Elemtype X, AvlTree T);
//Position FindMin(AvlTree T);
//Position FindMax(AvlTree T);

AvlTree Insert(Elemtype X, AvlTree T);
//AvlTree Delete(Elemtype X, AvlTree T);

//Elemtype Retrieve(Position P);

 avl.cpp

#include "avl.h"
#include<cstdio>
#include<cstdlib>
#include<cassert>
#include<algorithm>
struct AvlNode{
	Elemtype Element;
	AvlTree Left;
	AvlTree Right;
	int Height;
};

static int Height(Position p){
	if (p == nullptr)
		return -1;
	else
		return p->Height;
}
static Position SingleRotateWithLeft(Position K2){
	Position K1;
	K1 = K2->Left;
	K2->Left = K1->Right;
	K1->Right = K2;

	K2->Height = std::max(Height(K2->Left), Height(K2->Right)) + 1;
	K1->Height = std::max(Height(K1->Left), Height(K1->Right)) + 1;

	return K1; /*New root*/
}
static Position SingleRotateWithRight(Position K2){
	Position K1;
	K1 = K2->Right;
	K2->Right = K1->Left;
	K1->Left = K2;

	K2->Height = std::max(Height(K2->Left), Height(K2->Right)) + 1;
	K1->Height = std::max(Height(K1->Left), Height(K1->Right)) + 1;

	return K1; /*New root*/
}
static Position DoubleRotateWithLeft(Position K3){
	K3->Left = SingleRotateWithRight(K3->Left);
	return SingleRotateWithLeft(K3);
}
static Position DoubleRotateWithRight(Position K3){
	K3->Right = SingleRotateWithLeft(K3->Right);
	return SingleRotateWithRight(K3);
}
AvlTree Insert(Elemtype X, AvlTree T){
	if (T == nullptr){
		/*Create and return a one-node tree*/
		T = (AvlTree)malloc(sizeof(struct AvlNode));
		assert(T != nullptr, "Out of Space");
		T->Element = X; T->Height = 0;
		T->Left = T->Right = nullptr;
	}
	else if (X < T->Element){
		T->Left = Insert(X, T->Left);
		if (Height(T->Left) - Height(T->Right) == 2){
			if (X < T->Left->Element){
				T = SingleRotateWithLeft(T);
			}
			else{
				T = DoubleRotateWithLeft(T);
			}
		}
	}
	else if (X > T->Element){
		T->Right = Insert(X, T->Right);
		if (Height(T->Right) - Height(T->Left) == 2){
			if (X > T->Right->Element){
				T = SingleRotateWithRight(T);
			}
			else{
				T = DoubleRotateWithRight(T);
			}
		}
	}
	/*Else X is in the tree already; we'll do nothing */
	T->Height = std::max(Height(T->Left), Height(T->Right)) + 1;
	return T;
}