PTA Root of AVL Tree --- 平衡二叉树的建立

昨天是种树节，今天才种完这个平衡二叉树。。。种这个树也太细节了

思路：

1. 种树时记录每个结点的Height，通过这个可以计算每个树结点的平衡因子：通过左子树Height减右子树的Height。（注意：我的Height在叶结点时为1，之后每向上一层加1）
2. 对于不平衡的情况分为
   蓝色为插入结点，图来自MOOC中浙江大学老师的PPT
   SingleLeftRotation：
   ingleRightRotation：
   DoubleLeftRightRotation：
   
   DoubleRightLeftRotation：
   这四种解决办法在函数中实现
3. Height： 当插入某个元素时，要做的操作：1.插入时要插入到叶结点中，即需要创建一个新的结点，要初始化这个叶结点的Height 2.创建新的结点后需要更新每个结点的Height
   在代码中两种情况统一调用一句代码
4. 平衡因子：对于不平衡的二叉树，可能平衡因子绝对值大于等于二的结点有很多，但只需要先调整最深层（也就是最靠下的）的这个结点
   这就看到递归的好处，调用插入函数时，先从根往叶结点找插入元素的合适位置；当插入后就一步一步退出递归，也即从叶结点往根一步一步更新每个结点的Height，并且找不满足平衡规则的结点
   这样就使得找到的不满足平衡结点是最深的结点

#include <stdio.h>
#include <stdlib.h>

typedef struct TREE {
	int v;
	struct TREE* Left;
	struct TREE* Right;
	int Heigh; //以叶结点记为为第一层
}* Tree;

int GetHeigh(Tree T) {
	if (T) //T指向的结点不为空
		return T->Heigh;
	else
		return 0;
}

int Max(int x, int y) { //更新树高时用到
	return x > y ? x : y;
}

/*进行左旋*/
Tree SingleLeftRotation(Tree T) {
	Tree B = T->Left;
	T->Left = B->Right;
	B->Right = T;
	T->Heigh = Max(GetHeigh(T->Left), GetHeigh(T->Right)) + 1;//更新树高 树高是由结点的子结点的高决定
	B->Heigh = Max(GetHeigh(B->Left), GetHeigh(B->Right)) + 1;
	return B; //返回新的父节点结点
}

/*进行右旋*/
Tree SingleRightRotation(Tree T) {
	Tree B = T->Right;
	T->Right = B->Left;
	B->Left = T;
	T->Heigh = Max(GetHeigh(T->Left), GetHeigh(T->Right)) + 1;//更新树高 树高是由结点的子结点的高决定
	B->Heigh = Max(GetHeigh(B->Left), GetHeigh(B->Right)) + 1;
	return B; //返回新的父节点结点
}

/*进行左-右旋*/
Tree DoubleLeftRightRotation(Tree T) { //记T左子为B B右子为C
	/* 将B与C做右单旋，C被返回 */
	T->Left = SingleRightRotation(T->Left);
	/* 将A与C做左单旋，C被返回 */
	return SingleLeftRotation(T);
}

/*进行右-左旋*/
Tree DoubleRightLeftRotation(Tree T) { //记T子为B B左子为C
	/* 将B与C做左单旋，C被返回 */
	T->Right = SingleLeftRotation(T->Right);
	/* 将A与C做右单旋，C被返回 */
	return SingleRightRotation(T);
}

/*重点注意   像创建结点，为了平衡改变父节点等操作，需要返回这个结点或者新的父节点，在调用时用一个合适的指针接受这些返回的值*/
/*对于不平衡的二叉树，可能平衡因子绝对值大于等于二的结点有很多，但只需要先调整最深层的这些结点*/
Tree Insert(Tree T, int V) {
	if (!T) { //T指向的结点为空
		T = (Tree)malloc(sizeof(struct TREE));
		T->v = V;	T->Left = T->Right = NULL;	//在下面有统一更新树高的代码
		//return T; //将建立新的结点返回
	}
	else { //注意插入时会有不平衡的情况
		
		if (V < T->v) { //应该往左子节点下插入
			T->Left = Insert(T->Left, V);
			if (GetHeigh(T->Left) - GetHeigh(T->Right) == 2) { //插入后左子结点的层数高
				
				if (V < T->Left->v) { //此时比左子节点的值小 应该左旋 
					T = SingleLeftRotation(T);
				}
				else { //比左子结点值大 应该左-右旋
					T = DoubleLeftRightRotation(T);
				}
			}
		}
		else if (V > T->v) { //应该往右子结点下插入
			T->Right = Insert(T->Right, V);
			if (GetHeigh(T->Left) - GetHeigh(T->Right) == -2) { //插入后右结点层数高
				if (V > T->Right->v) { //值大于右结点 应该右旋
					T = SingleRightRotation(T);
				}
				else { //比左结点大 应该右-左旋
					T = DoubleRightLeftRotation(T);
				}
			}
		}
	}
	/*更新树高 从递归最后一层反向往前更新*/
	T->Heigh = Max(GetHeigh(T->Left), GetHeigh(T->Right)) + 1; //统一更新树高
	return T;  //返回创建的结点或者新的父节点
}

int main() {
	int N, V;
	scanf("%d", &N);
	Tree T = NULL;
	for (int i = 0; i < N; i++) {
		scanf("%d", &V);
		T = Insert(T, V);
	}
	printf("%d", T->v);
}