Go: LeetCode簡單題,簡單做(sort.Search)


前言

正值端午佳節,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.通常用來在有序列表中進行過濾,也可以判斷某個元素是否存在


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM