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

最新推荐文章于 2022-07-31 12:11:08 发布

Jinhua_Wei

最新推荐文章于 2022-07-31 12:11:08 发布

阅读量1.7k

点赞数

分类专栏： golang 数据结构文章标签： golang 数据结构

本文链接：https://blog.csdn.net/Jinhua_Wei/article/details/79595507

版权

本文介绍了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
//         /            左          /     右