在Go中对切片进行二分查找需确保数据有序,sort包提供sort.Search实现灵活查找,通过条件函数定位首个不小于目标的索引,结合预定义函数如sort.SearchInts、sort.SearchStrings可简化操作,还可利用插入点保持有序。

在Go语言中,对切片进行二分查找时,必须保证数据已排序。标准库
sort
提供了高效且类型安全的工具,能快速实现查找、插入等操作。以下是实际使用中的常见场景和方法。
使用 sort.Search 进行自定义二分查找
sort.Search
是最灵活的方式,适用于任意有序切片。它接受长度 n 和一个判断条件 f(i),返回满足 f(i) 为 true 的最小索引。
例如:在一个升序整数切片中查找目标值的位置:
func binarySearch(arr []int, target int) int { i := sort.Search(len(arr), func(i int) bool { return arr[i] >= target }) if i < len(arr) && arr[i] == target { return i } return -1 // 未找到}
这个写法的关键在于条件函数
arr[i] >= target
,它定位第一个不小于目标值的位置,再通过额外判断确认是否相等。
使用预定义函数简化查找
对于常见类型,
sort
包提供了专用函数,代码更简洁:
立即学习“go语言免费学习笔记(深入)”;
sort.Ints(arr)
—— 对整型切片排序
sort.Strings(arr)
—— 对字符串切片排序
sort.SearchInts(arr, x)
—— 在已排序整型切片中查找 x
sort.SearchStrings(arr, x)
—— 在已排序字符串切片中查找 x示例:快速查找字符串是否存在
names := []string{"Alice", "Bob", "Charlie"}sort.Strings(names)index := sort.SearchStrings(names, "Bob")if index != len(names) && names[index] == "Bob" { fmt.Println("Found at", index)}
插入新元素并保持有序
利用
sort.Search
找到插入点,可将新元素放入正确位置而不破坏顺序。
比如向有序整数切片插入一个数:
func insertSorted(arr []int, x int) []int { i := sort.Search(len(arr), func(i int) bool { return arr[i] >= x }) arr = append(arr, 0) copy(arr[i+1:], arr[i:]) arr[i] = x return arr}
这段代码先用
Search
定位插入索引,然后扩展切片并移动后续元素。
基本上就这些。只要数据有序,
sort.Search
和配套函数就能高效完成查找与维护。关键是理解条件函数的语义:找“第一个满足 >= 目标”的位置。掌握这一点,各种二分操作都容易推导。
以上就是Golang sort/search切片二分查找实践的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1408286.html
微信扫一扫
支付宝扫一扫