A1066 Root of AVL Tree (25) 平衡二叉树的一般结题思路

注意点，在更新L和R操作中，应先更新root（此时为低位置）的高度，再更新temp（此时为高位置）的高度，
由于本题输入的数据有限。可以考虑测试相关数据，因为输出的只有最顶端的数据，根据平衡二叉树的性质，此时的节点必定为中间的数值。先排序解决

#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=110;
struct node
{
	int data,height;
	node *lchild,*rchild;
};
int getHeight(node *root)
{
	if(root==NULL)
	return 0;
	return root->height;
}
int getBalanceFactor(node *root)
{
	return getHeight(root->lchild)-getHeight(root->rchild);
}
void updateHeight(node* root)
{
	root->height=max(getHeight(root->lchild),getHeight(root->rchild))+1;
}
void L(node* &root)
{
	node *temp=root->rchild;
	root->rchild=temp->lchild;
	temp->lchild=root;
	updateHeight(root);
	updateHeight(temp);
	root=temp;
}
void R(node* &root)
{
	node *temp=root->lchild;
	root->lchild=temp->rchild;
	temp->rchild=root;
	updateHeight(root);
	updateHeight(temp);
	root=temp;
}
void insert(node* &root,int v)
{
	if(root==NULL)
	{
	   root=new node;
	   root->data=v;
	   root->lchild=root->rchild=NULL;
	   root->height=1;
	   return;
	}
	if(v>=root->data)
	{
		insert(root->rchild,v);
		updateHeight(root);
		if(getBalanceFactor(root)==-2)
		{
			if(getBalanceFactor(root->rchild)==-1)
			{
				L(root);
			}
			else if(getBalanceFactor(root->rchild)==1)
			{
				R(root->rchild);
			    L(root);
			}
		}
	}
	else
	{
		insert(root->lchild,v);
		updateHeight(root);
		if(getBalanceFactor(root)==2)
		{
			if(getBalanceFactor(root->lchild)==1)
			{
				R(root);
			}
			else if(getBalanceFactor(root->lchild)==-1)
			{
				L(root->lchild);
				R(root);
			}
		}
	}
}
int main()
{
  int n,index;
  cin>>n;
  node *root=NULL;
  for(int i=0;i<n;i++)
  {
  	scanf("%d",&index);
  	insert(root,index);
  }
  printf("%d",root->data);
}