(c语言)AVL平衡二叉树基本小白操作集(附测试用例)

本试验取材于浙江大学mooc《数据结构》，包括基本的平衡二叉树的操作集。本次代码用了好久才把它调试出来，如果原理不明白的话，可以看何老师讲的课程，b站也有哟！
平衡二叉树是二叉搜索树的再一次改进，尤其那几种旋转还是比较拗口。
大家可以参考这篇文章。之前看他的深有启发
下面就给出一般的对平衡二叉树的操作：

//要写写自己家的二叉搜索树
#include<stdio.h>
#include<stdlib.h>
typedef struct AVLNode *Position;
typedef Position AVLNodea; //AVL树类型
typedef int ElmentType;
typedef AVLNodea AVLTree;
struct AVLNode{
	ElmentType Data;//结点数据
	AVLTree Left;//指向左子树
	AVLTree Right;//指向右子树
	int Height;//树高
};

int Max(int a,int b)
{
	return a>b?a:b;
}
int GetHeight(AVLTree BT){
	int HL,HR,MaxH;
	if(BT){
		HL = GetHeight(BT->Left);//递归求左子树的高度
		HR = GetHeight(BT->Right);//递归求右子树的高度
		MaxH = HL > HR ? HL:HR;//取左右子树较大的高度
		return (MaxH+1);//返回树的高度
	}
	else return 0;
}
AVLTree SingleLeftRotation(AVLTree A)
{
	//注意：A必须有一个左孩子结点B
	//将A与B做左单旋，更新A与B的高度，返回新的根结点B；
	AVLTree B = A->Left;
	A->Left = B->Right;
	A->Right = A;
	A->Height = Max(GetHeight(A->Left),GetHeight(A->Right))+1;
	B->Height = Max(GetHeight(B->Left),GetHeight(B->Right))+1;
	
	return B;
}//程序完成了左单选
AVLTree SingleRightRotation(AVLTree A)
{
	//注意：A必须有一个右孩子结点B
	//将A与B做左=右单旋，更新A与B的高度，返回新的根结点B；
	AVLTree B = A->Right;
	A->Right = B->Left;
	A->Left = A;
	A->Height = Max(GetHeight(A->Left),GetHeight(A->Right))+1;
	B->Height = Max(GetHeight(B->Left),GetHeight(B->Right))+1;
	
	return B;
}//程序完成了右单选
AVLTree DoubleLeftRightRotation(AVLTree A)
{
	//注意：A必须有一个左子结点B,且B必须有一个右子结点C
	//将A、B与C做两次单旋，返回新的根结点C
	
	//将B与C做右单选，C被返回
	A->Left = SingleRightRotation(A->Left);
	//将A与C做左单旋，C被返回
	return SingleLeftRotation(A);
}
AVLTree DoubleRightLeftRotation(AVLTree A)
{
	//注意：A必须有一个右子结点B,且B必须有一个左子结点C
	//将A、B与C做两次单旋，返回新的根结点C
	A->Right = SingleLeftRotation(A->Right);
	//将A与C做左单旋，C被返回
	return SingleRightRotation(A);
}
AVLTree Insert(AVLTree T,ElmentType X)
{

	//将X插入AVL树T中，并且返回调整后的AVL树
	if(!T){//若插入空树，则新建包含一个结点的树
		T = (AVLTree)malloc(sizeof(struct AVLNode));
		T->Data = X;
		T->Height = 0;
		T->Left = T->Right = NULL;
	}//if(插入空树结束)
	else if(X<T->Data){
		//插入T的左子树
		T->Left = Insert(T->Left,X);
		//如果需要左旋
		if(GetHeight(T->Left)-GetHeight(T->Right)== 2)
			if(X<T->Left->Data)
				T = SingleLeftRotation(T);
			else
				T = DoubleLeftRightRotation(T);
	}	
	else if(X>T->Data){
		//插入T的右子树
		T->Right = Insert(T->Right,X);
		//如果需要右旋
		if(GetHeight(T->Left)-GetHeight(T->Right)==-2)
			if(X>T->Right->Data)
				T = SingleRightRotation(T);
			else
				T = DoubleRightLeftRotation(T);
	}
	//别忘了跟新树高
	T->Height = Max(GetHeight(T->Left),GetHeight(T->Right))+1;
	return T;
}
void InorderTraversal( AVLTree BT )
{
	
    if( BT ) {
		
        InorderTraversal( BT->Left );
        /* 此处假设对BT结点的访问就是打印数据 */
         /* 假设数据为整型 */
         printf("%d ", BT->Data);
        InorderTraversal( BT->Right );
    }
}
int main()
{
	AVLTree root = (AVLTree)malloc(sizeof(struct AVLNode));
	root->Left=root->Right=NULL;
	root->Height = 0;
	root->Data = 13;
	root = Insert(root,26);
	root = Insert(root,9);	
	root = Insert(root,24);
	root = Insert(root,34);
	root = Insert(root,1);
	root = Insert(root,113);

		
	printf("inorder traverse\n");
	InorderTraversal(root);
	return 0;
}

总结：测试用例也很直白，采用了，一个个insert的方式，直观显然，最后采用中序遍历直接出遍历结果。