PAT A1066

PAT A1066

题意

AVL树是自平衡二叉搜索树，在AVL树中，任何节点的两个子树的高度最多相差一个。如果在任何时候它们相差多于⼀个，则重新平衡以恢复此属性。 图中说明了旋转规则，现在给出⼀一系列插入序列，要求输出根节点的值。

分析

写出构建AVL（衡二叉搜索树）的代码模版即可。
//生成一个新结点，v为结点权值
node *newNode(int v)

//获取以root为根结点的子树的当前height
int getHeight(node* root)

/计算结点root的平衡因子
int getBalanceFactor(node* root)

//更新结点root的height
void updateHeight(node* root)

//左旋（Left Rotation）
void L(node* &root)

//右旋（Right Rotation）
void R(node* &root)

//插入权值为v的结点
void insert(node* &root, int v)

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

struct node {
	int v, height;//v为结点权值，height为当前子树高度 
	node *lchild, *rchild;//左右孩子结点地址
}; 
//生成一个新结点，v为结点权值
node *newNode(int v)
{
	node* Node = new node;
	Node->v = v;
	Node->height = 1;//结点高度初始为1
	Node->lchild = Node->rchild = NULL;
	return Node;
}
//获取以root为根结点的子树的当前height
int getHeight(node* root)
{
	if(root == NULL)
	{
		return 0;
	}
	return root->height;
} 
//计算结点root的平衡因子
int getBalanceFactor(node* root)
{
	return getHeight(root->lchild) - getHeight(root->rchild);
}
//更新结点root的height
void updateHeight(node* root)
{
	root->height = max(getHeight(root->lchild), getHeight(root->rchild)) + 1;
}

//左旋（Left Rotation）
void L(node* &root)
{
	node* temp = root->rchild;
	root->rchild = temp->lchild;
	temp->lchild = root;
	updateHeight(root);
	updateHeight(temp);
	root = temp;
}
//右旋（Right Rotation）
void R(node* &root)
{
	node* temp = root->lchild;
	root->lchild = temp->rchild;
	temp->rchild = root;
	updateHeight(root);
	updateHeight(temp);
	root = temp;
}
//插入权值为v的结点
void insert(node* &root, int v)
{
	if (root == NULL)
	{
		root = newNode(v);
		return;
	}
	if (v < root->v)
	{
		insert(root->lchild, v);
		updateHeight(root);
		if (getBalanceFactor(root) == 2)
		{
			if (getBalanceFactor(root->lchild) == 1)//LL 
			{
				R(root); 
			}
			else if (getBalanceFactor(root->lchild) == -1)//LR
			{
				L(root->lchild);
				R(root);
			}
		}
	}
	else
	{
		insert(root->rchild, v);
		updateHeight(root);
		if (getBalanceFactor(root) == -2)
		{
			if (getBalanceFactor(root->rchild) == -1)//RR
			{
				L(root); 
			}
			else if (getBalanceFactor(root->rchild) == 1)//RL
			{
				R(root->rchild);
				L(root);
			}
		}
	}
}
int main()
{
	int n;
	cin >> n;
	node *root = NULL;
	for (int i = 0; i < n; i++)
	{
		int x;
		cin >> x;
		insert(root, x);
	}
	cout << root->v << endl;
	return 0;
}