golang sort.SearchInts查找算法

源码

// SearchInts 在排序的整数切片中搜索 x 并返回 Search 指定的索引。 
// 如果 x 不存在，则返回值是插入 x 的索引（它可能是 len(a)）。 切片必须按升序排序。
func SearchInts(a []int, x int) int {
	return Search(len(a), func(i int) bool { return a[i] >= x })
}
func Search(n int, f func(int) bool) int {
	// 定义 f(-1) == false and f(n) == true.
	// 如果 f(i-1) == false, 则f(j) == true.
	i, j := 0, n
	for i < j {
		h := int(uint(i+j) >> 1) // 转换成uint防止溢出
		// i ≤ h < j
		if !f(h) {
			i = h + 1 // 保证f(i-1) == false
		} else {
			j = h // 保证 f(j) == true
		}
	}
	// 此时i == j, f(i-1) == false, f(j) == f(i) == true 得出 answer is i.
	return i
}

SearchInts方法不会在找到元素之后立刻返回，而是继续将左/右游标继续移动直到相等，以此实现目标不存在时返回要插入的下标。