昨天是种树节,今天才种完这个平衡二叉树。。。种这个树也太细节了
思路:
- 种树时记录每个结点的Height,通过这个可以计算每个树结点的平衡因子:通过左子树Height减右子树的Height。(注意:我的Height在叶结点时为1,之后每向上一层加1)
- 对于不平衡的情况分为
蓝色为插入结点,图来自MOOC中浙江大学老师的PPT
SingleLeftRotation:
ingleRightRotation:
DoubleLeftRightRotation:
DoubleRightLeftRotation:
这四种解决办法在函数中实现 - Height: 当插入某个元素时,要做的操作:1.插入时要插入到叶结点中,即需要创建一个新的结点,要初始化这个叶结点的Height 2.创建新的结点后需要更新每个结点的Height
在代码中两种情况统一调用一句代码 - 平衡因子:对于不平衡的二叉树,可能平衡因子绝对值大于等于二的结点有很多,但只需要先调整最深层(也就是最靠下的)的这个结点
这就看到递归的好处,调用插入函数时,先从根往叶结点找插入元素的合适位置;当插入后就一步一步退出递归,也即从叶结点往根一步一步更新每个结点的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);
}