golang 红黑树map实现_golang treemap-优快云博客

本文链接：https://blog.youkuaiyun.com/sky_terra/article/details/106695637

本文详细探讨了如何在Golang中利用红黑树数据结构实现高效的映射操作，涵盖了红黑树的基本原理、插入、删除和查找算法，并通过实例展示了具体的代码实现，帮助读者理解其在Go编程中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//------------------------------------------------------------------------------
// rbtree.go
// 红黑树
//------------------------------------------------------------------------------

package rbmap

const (
	RED   bool = true
	BLACK bool = false
)

type node struct {
	key   int64
	value interface{}

	left   *node
	right  *node
	parent *node

	color bool
}

var nilnode = &node{
	color: BLACK,
}

type rbtree struct {
	root *node
}

// 搜索key
func search(x *node, k int64) *node {
	if x == nilnode || k == x.key {
		return x
	}

	if k < x.key {
		return search(x.left, k)
	} else {
		return search(x.right, k)
	}
}

// 最小key元素
func maximum(x *node) *node {
	for x.left != nilnode {
		x = x.left
	}
	return x
}

// 最大key元素
func minimum(x *node) *node {
	for x.right != nilnode {
		x = x.right
	}
	return x
}

// 结点x的后继及结点
func successor(x *node) *node {
	if x.right != nilnode {
		return minimum(x)
	}

	y := x.parent

	for y != nilnode && x == y.right {
		x = y
		y = y.parent
	}

	return y
}

func remove(T *rbtree, z *node) *node {
	x := nilnode
	y := nilnode

	if z.left == nilnode || z.right == nilnode {
		y = z
	} else {
		y = successor(z)
	}

	if y.left != nilnode {
		x = y.left
	} else {
		x = y