sort.Search用于在有序序列中二分查找首个满足条件的索引,其核心是构造返回bool的函数f,例如查找目标值时判断“大于等于”,再验证该位置元素是否相等,从而实现O(log n)高效搜索。

在Golang中,sort.Search 是一个非常高效的查找工具,用于在已排序的序列中查找满足某个条件的第一个位置。它基于二分查找实现,时间复杂度为 O(log n)。虽然它不像 sort.SearchInts 那样直接“查找某个值”,但它更灵活,适用于各种自定义查找场景。
理解 sort.Search 的函数签名
func Search(n int, f func(int) bool) int
这个函数接收两个参数:
- n:搜索范围的长度(比如切片的长度)
- f:一个函数,判断索引 i 是否满足某个条件(通常是“大于等于目标值”)
它返回第一个使 f(i) 为 true 的索引。如果不存在,则返回 n。
立即学习“go语言免费学习笔记(深入)”;
使用 sort.Search 查找特定元素
假设你有一个已排序的整数切片,想查找某个值是否存在,并获取其索引。
步骤如下:
- 确保切片已经排序(升序)
- 使用
sort.Search找到第一个大于等于目标值的位置 - 检查该位置的值是否等于目标值
示例代码:
package main
import (
"fmt"
"sort"
)
func main() {
nums := []int{1, 3, 5, 7, 9, 11}
target := 7
// 使用 sort.Search 查找第一个 >= target 的索引
i := sort.Search(len(nums), func(i int) bool {
return nums[i] >= target
})
// 检查是否找到目标值
if i < len(nums) && nums[i] == target {
fmt.Printf("找到目标值 %d,索引为 %d\n", target, i)
} else {
fmt.Printf("未找到目标值 %d\n", target)
}
}
封装成通用查找函数
你可以将上述逻辑封装成一个函数,便于复用:
func findInSortedSlice(nums []int, target int) int {
i := sort.Search(len(nums), func(i int) bool {
return nums[i] >= target
})
if i < len(nums) && nums[i] == target {
return i
}
return -1 // 表示未找到
}
调用 findInSortedSlice(nums, 7) 就能返回索引或 -1。
适用于其他类型和结构体
只要数据是有序的,sort.Search 也能用于字符串、浮点数甚至结构体。
例如,查找按姓名排序的用户列表:
type User struct {
Name string
}
users := []User{{"Alice"}, {"Bob"}, {"Charlie"}}
target := "Bob"
i := sort.Search(len(users), func(i int) bool {
return users[i].Name >= target
})
if i < len(users) && users[i].Name == target {
fmt.Println("找到用户:", users[i].Name)
}
基本上就这些。关键是要构造好那个返回 bool 的函数,让它表达“从哪里开始满足条件”。只要理解这一点,sort.Search 就很容易掌握。注意:输入数据必须是有序的,否则结果不可预测。










