前言
正值端午佳節,LeetCode也很懂。這兩天都是簡單題,早點做完去包粽子。
故事從一道簡單題說起
你是產品經理,目前正在帶領一個團隊開發新的產品。不幸的是,你的產品的最新版本沒有通過質量檢測。由於每個版本都是基於之前的版本開發的,所以錯誤的版本之后的所有版本都是錯的。
假設你有 n 個版本 [1, 2, ..., n],你想找出導致之后所有版本出錯的第一個錯誤的版本。
你可以通過調用 bool isBadVersion(version) 接口來判斷版本號 version 是否在單元測試中出錯。實現一個函數來查找第一個錯誤的版本。你應該盡量減少對調用 API 的次數。
示例:
給定 n = 5,並且 version = 4 是第一個錯誤的版本。
調用 isBadVersion(3) -> false
調用 isBadVersion(5) -> true
調用 isBadVersion(4) -> true
所以,4 是第一個錯誤的版本。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/first-bad-version
簡單題,所以重拳出擊,簡單分析:產品版本號是1到n,如果每個版本都調用一個isBadVersion
函數,將結果放入數組中,則為:[false, ..., false, ture, ..., ture]
,即我們從左到右找到第一個值為true
的下標即為答案,可使用二分法
去找第一個滿足isBadVersion
函數值為true的version。
func firstBadVersion(n int) int {
left, right := 1, n
for ; left < right; {
mid := left + (right-left) / 2
if isBadVersion(mid) {
right = mid
} else {
left = mid + 1
}
}
return left
}
PASS,查看題解思路一致。然而,再看一下官方的Go版,只用了一行~
func firstBadVersion(n int) int {
return sort.Search(n, func(version int) bool { return isBadVersion(version) })
}
sort.Search是何方神聖
func Search(n int, f func(int) bool) int {
// Define f(-1) == false and f(n) == true.
// Invariant: f(i-1) == false, f(j) == true.
i, j := 0, n
for i < j {
h := int(uint(i+j) >> 1) // avoid overflow when computing h
// i ≤ h < j
if !f(h) {
i = h + 1 // preserves f(i-1) == false
} else {
j = h // preserves f(j) == true
}
}
// i == j, f(i-1) == false, and f(j) (= f(i)) == true => answer is i.
return i
}
Search
函數內部通過二分搜索
在下標范圍[0,n)
中去找到最小的下標是函數f(i)
為true,將其返回。要注意,如果下標范圍內不存在滿足條件的小標,函數返回n。
例:
package main
import (
"fmt"
"sort"
)
func main() {
a := []int{1, 2, 3, 4, 5}
i := sort.Search(len(a), func(i int) bool { return a[i] >= 3 }) // 閉包
fmt.Println(i) // 2
// 尋找某個值
fmt.Println(FindX(a, 4))
fmt.Println(FindX(a, 100))
}
func FindX(sortedA []int, x int) int {
i := sort.Search(len(sortedA), func(i int) bool { return sortedA[i] >= x })
if i < len(sortedA) && sortedA[i] == x {
return i
}
return -1
}
sort.Search總結
1.訪問的下標范圍: [0,n)
2.如果不存在滿足調節,函數返回n
3.通常用來在有序列表中進行過濾,也可以判斷某個元素是否存在