【golang/方法记录】有序列表的简易实现

最新推荐文章于 2025-01-12 09:20:54 发布

o0o_-_

最新推荐文章于 2025-01-12 09:20:54 发布

阅读量1k

点赞数 1

分类专栏： Go 文章标签： golang 开发语言后端

本文链接：https://blog.csdn.net/qq_33446100/article/details/123434855

版权

Go 专栏收录该内容

19 篇文章 0 订阅

订阅专栏

说在前面

go version：go1.14.1 windows/amd64

实现

借助golang中的sort包可以方便的使用二分查找。

func Search(n int, f func(int) bool) int {
	// Define f(-1) == false and f(n) == true.
	// Invariant: f(i-1) == false, f(j) == true.
	i, j := 0, n
	for i < j {
		h := int(uint(i+j) >> 1) // avoid overflow when computing h
		// i ≤ h < j
		if !f(h) {
			i = h + 1 // preserves f(i-1) == false
		} else {
			j = h // preserves f(j) == true
		}
	}
	// i == j, f(i-1) == false, and f(j) (= f(i)) == true  =>  answer is i.
	return i
}

借助二分查找可以方便的实现有序列表。

代码

例如有序的uint64切片

func SeachUint64List(arr []uint64, x uint64) int {
	return sort.Search(len(arr), func(i int) bool {
		return arr[i] >= x
	})
}

func InsertOrderedUint64List(arr []uint64, x uint64) []uint64 {
	i := SeachUint64List(arr, x)

	arr = append(arr, 0)
	copy(arr[i+1:], arr[i:])
	arr[i] = x

	return arr
}

func main() {
	var arr []uint64

	arr = InsertOrderedUint64List(arr, 12)
	arr = InsertOrderedUint64List(arr, 2)
	arr = InsertOrderedUint64List(arr, 102)
	arr = InsertOrderedUint64List(arr, 1)
	arr = InsertOrderedUint64List(arr, 14542)
	arr = InsertOrderedUint64List(arr, 112)
	arr = InsertOrderedUint64List(arr, 142)

	fmt.Println(arr)
	// [1 2 12 102 112 142 14542]
}

其他的可以通过定义结构体，并使用Key进行排序，例如

type Node struct {
	Key   uint64
	Value string
}

func SeachNodeList(arr []*Node, x uint64) int {
	return sort.Search(len(arr), func(i int) bool {
		return arr[i].Key >= x
	})
}

func InsertOrderedNodeList(arr []*Node, x *Node) []*Node {
	if x == nil {
		return arr
	}

	i := SeachNodeList(arr, x.Key)

	arr = append(arr, &Node{})
	copy(arr[i+1:], arr[i:])
	arr[i] = x

	return arr
}

func main() {
	var arr []*Node

	arr = InsertOrderedNodeList(arr, &Node{Key: 12, Value: "ha"})
	arr = InsertOrderedNodeList(arr, &Node{Key: 1432, Value: "ta"})
	arr = InsertOrderedNodeList(arr, &Node{Key: 112, Value: "see"})
	arr = InsertOrderedNodeList(arr, &Node{Key: 1452, Value: "sight"})
	arr = InsertOrderedNodeList(arr, &Node{Key: 13332, Value: "peer"})
	arr = InsertOrderedNodeList(arr, &Node{Key: 12, Value: "one"})
	arr = InsertOrderedNodeList(arr, &Node{Key: 2, Value: "kiss"})

	for _, node := range arr {
		fmt.Println(*node)
	}
	
	// {2 kiss}
	// {12 one}
	// {12 ha}
	// {112 see}
	// {1432 ta}
	// {1452 sight}
	// {13332 peer}
}