Go语言中sort.Search()的使用方法（数组中通过值来取索引）

最新推荐文章于 2025-02-11 10:18:56 发布

ppingfann

最新推荐文章于 2025-02-11 10:18:56 发布

阅读量1.1w

点赞数 10

CC 4.0 BY-SA版权

分类专栏： Golang 文章标签： go

本文链接：https://blog.youkuaiyun.com/hty46565/article/details/109689515

Golang 专栏收录该内容

5 篇文章

订阅专栏

Go语言的sort.Search()函数通过二分查找在已排序的数组中找到符合条件的最小索引。本文详细介绍了其使用模板、功能及常见应用场景，并提供了两个示例，演示如何在整数数组和字符串数组中找到满足特定条件的索引值。

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

方法名

sort.Search()

使用模板

index := sort.Search(n int,f func(i int) bool) int

主要功能

该函数使用二分查找的方法，会从[0, n)中取出一个值index，index为[0, n)中最小的使函数f(index)为True的值，并且f(index+1)也为True。
如果无法找到该index值，则该方法为返回n。

常用场景

该方法一般用于从一个已经排序的数组中找到某个值所对应的索引。
或者从字符串数组中，找到满足某个条件的最小索引值，比如etcd中的键值范围查询就用到了该方法。

具体例子

看该方法的描述会比较抽象，我们可以看两个具体的例子：

func main() {
    a := []int{1,2,3,4,5}
    d := sort.Search(len(a), func(i int) bool { return a[i]>=3})
    fmt.Println(d)
}

执行结果：2

    a := []byte("a")
    b := []byte("ab")
    c := []byte("ac")
    d := []byte("acd")
    e := []byte("ad")

    f := []byte("abc")
    g := [][]byte{a, b, c, d, e}
    h := sort.Search(len(g), func(i int) bool { return bytes.Compare(g[i], f)>=0})
    fmt.Println(h)

执行结果：2