在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)
—— 在已排序整型切片中查找 xsort.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和配套函数就能高效完成查找与维护。关键是理解条件函数的语义:找“第一个满足 >= 目标”的位置。掌握这一点,各种二分操作都容易推导。










