PAT——1066 Root of AVL Tree（内含注释）

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805404939173888

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef struct node* avl;
struct node{
	int data;
	avl left,right;
};
int GetHeight(avl a)//获取高度 
{
	if(!a) return 0;
	else
	{
		int l=GetHeight(a->left)+1;
		int r=GetHeight(a->right)+1;
		return l>r?l:r;
	}
}
avl L(avl a)//左旋转 
{
	avl b=a->right;
	a->right=b->left;
	b->left=a;
	return b;
}
avl R(avl a)//右旋转 
{
	avl b=a->left;
	a->left=b->right;
	b->right=a;
	return b;
}
avl LR(avl a)//左右旋转 
{
	a->left=L(a->left);
	return R(a);
}
avl RL(avl a)//右左旋转 
{
	a->right=R(a->right);
	return L(a);
}
avl Insert(avl a,int tmp)//插入节点 
{
	if(!a)
	{
		avl a=(avl)malloc(sizeof(struct node));//必须要为a开辟空间 
		a->data=tmp;
		a->left=a->right=NULL;
		return a; 
		//这里的return a一定要写，因为新开辟的a是局部变量，在if判断后就被剥夺空间了，所以必须在它的作用域内返回 
	}
	else if(tmp<a->data)//平衡二叉树是二叉排序树，左小右大 
	{
		a->left=Insert(a->left,tmp);
		if(GetHeight(a->left)-GetHeight(a->right)==2)
		{
			if(tmp<a->left->data) a=R(a);
			else a=LR(a);
		}
	}
	else //平衡二叉树是二叉排序树，左小右大 
	{
		a->right=Insert(a->right,tmp);
		if(GetHeight(a->left)-GetHeight(a->right)==-2)
		{
			if(tmp>a->right->data) a=L(a);
			else a=RL(a);
		}
	}
	return a;
}
int main()
{
	int n,tmp;
	cin>>n;
	avl a;//这个NULL的赋值也要写，否则会出现段错误，因为要为a开辟空间 
	while(n--)
	{
		cin>>tmp;
		a=Insert(a,tmp);
	}
	cout<<a->data;
}

说明

建议不了解AVL是什么的同学先看看这篇文章——漫话：什么是平衡(AVL)树？这应该是把AVL树讲的最好的文章了，主要看数据结构的讲解部分，代码看我这里的就行

当你了解之后再结合刚才的理论看左旋，右旋，左右旋，右左旋对应的代码，相信能更好地帮助你理解

tips：在理解过程中适当作图是一个不错的选择