什么是平衡二叉树?
搜索树结点不同插入次序,将导致不同的深度和平均查找长度ASL,而平衡二叉树就是一颗任意一个结点的左右子树高度的差的绝对值不超过1的搜索二叉树。
附一个大佬写的图解:图解平衡二叉树
平衡二叉树的四种基本情况和操作:
- 左-左型:右旋
- 右-右型:左旋
- 左-右型:先左旋再右旋
- 右-左型:先右旋再左旋
struct AvlNode {
int data;
AvlNode lchild;
AvlNode rchild;
int height;//记录结点的高度
}
int Height(AvlNode T)
{
if(!T)return -1;
else return T.height
}
//左左型,右旋操作
AvlNode R_Rotate(AvlNode K2)
{
AvlNode K1;
K1 = K2.lchild;
K2.lchild = K1.rchild;
K1.rchild = K2;
//重新计算节点的高度
K2.height = Math.max(height(K2.lchild), height(K2.rchild)) + 1;
K1.height = Math.max(height(K1.lchild), height(K1.rchild)) + 1;
return K1;
}
//右右型,进行左旋
AvlNode L_Rotate(AvlNode K2) {
AvlNode K1;
K1 = K2.rchild;
K2.rchild = K1.lchild;
k1.lchild = K2;
//重新计算节点的高度
K2.height = Math.max(height(K2.lchild), height(K2.rchild)) + 1;
K1.height = Math.max(height(K1.lchild), height(K1.rchild)) + 1;
return K1;
}
//左-右型,进行左旋,再右旋
AvlNode L_R_Rotate(AvlNode K3) {
K3.lchild = L_Rotate(AvlNode K3.lchild);
return R_Rotate(AvlNode K3);
}
//右-左型,先进行右旋,再左旋
AvlNode R_L_Rotate(AvlNode K3) {
K3.rchild = R_Rotate(AvlNode K3.rchild);
return L_Rotate(AvlNode K3);
}
//插入数值操作
AvlNode insert(int X, AvlNode T) {
if (T == null) {
T = new AvlNode();
T.data = X;
T.lchild = T.rchild = null;
}
else if(X>T.data){//向右孩子递归插入
T.rchild = insert(X,T.rchild);
//插入完成后要进行调整
if (height(T.rchild) - height(T.lchild) == 2){
//右右型
if(X>T.rchlid.data)T = L_Rotate(T);
//右左型
else T = L_R_Rotate(T);
}
}
else if(X<T.data){//向左孩子递归插入
T.lchild = insert(X,T.lchild);
//插入完成后进行调整
if(X>T.lchild.data)T = R_L_Rotate(T);
else T = R_Rotate(T);
}
//重新计算T的高度
T.height = Math.max(height(T.lchild), height(T.rchild)) + 1;
return T;
}