golang实现高度平衡二叉树(AVL树)

本文介绍了AVL树作为高度平衡的二叉搜索树,其特点是防止搜索复杂度退化为线性。通过在插入和删除操作中进行平衡调整,确保树始终保持平衡。文章提供了Golang实现AVL树的源码,并展示了运行结果。
摘要由CSDN通过智能技术生成

AVL树简介

AVL 树是高度相对平衡(abs(height(node.lchild - height(node.rchild) < 2))的二叉搜索树。它和二叉搜索树主要的区别在AVL是高度平衡的,不会出现二叉搜索树极端情况:线性链表 搜索复杂度为N
实现上,AVL树在插入节点删除节点时要不断调整树,使其处在一个平衡状态。和二叉搜索树相比主要增加树旋转调整

实现源码

package main

import (
    "errors"
    "fmt"
)

var (
    errNotExist       = errors.New("index is not existed")
    errTreeNil        = errors.New("tree is null")
    errTreeIndexExist = errors.New("tree index is existed")
)

type Node struct {
    lchild *Node
    rchild *Node
    height int //树的深度
    index  int
    data   int
}

func max(data1 int, data2 int) int {
    if data1 > data2 {
        return data1
    }
    return data2
}

func getHeight(node *Node) int {
    if node == nil {
        return 0
    }
    return node.height
}

// 左旋转
//
//    node  BF = 2
//       \
//         prchild     ----->       prchild    BF = 1
//           \                        /   \
//           pprchild               node  pprchild
func llRotation(node *Node) *Node {
    prchild := node.rchild
    node.rchild = prchild.lchild
    prchild.lchild = node
    //更新节点 node 的高度
    node.height = max(getHeight(node.lchild), getHeight(node.rchild)) + 1
    //更新新父节点高度
    prchild.height = max(getHeight(prchild.lchild), getHeight(prchild.rchild)) + 1
    return prchild
}

// 右旋转
//             node  BF = -2
//              /
//         plchild     ----->       plchild    BF = 1
//            /                        /   \
//        pplchild                lchild   node

func rrRotation(node *Node) *Node {
    plchild := node.lchild
    node.lchild = plchild.rchild
    plchild.rchild = node
    node.height = max(getHeight(node.lchild), getHeight(node.rchild)) + 1
    plchild.height = max(getHeight(node), getHeight(plchild.lchild)) + 1
    return plchild
}

// 先左转再右转
//          node                  node
//         /            左          /     右
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值