LeetCode演算法題278:第一個錯誤的版本解析
阿新 • • 發佈:2018-11-20
你是產品經理,目前正在帶領一個團隊開發新的產品。不幸的是,你的產品的最新版本沒有通過質量檢測。由於每個版本都是基於之前的版本開發的,所以錯誤的版本之後的所有版本都是錯的。
假設你有 n 個版本 [1, 2, …, n],你想找出導致之後所有版本出錯的第一個錯誤的版本。
你可以通過呼叫 bool isBadVersion(version) 介面來判斷版本號 version 是否在單元測試中出錯。實現一個函式來查詢第一個錯誤的版本。你應該儘量減少對呼叫 API 的次數。
示例:
給定 n = 5,並且 version = 4 是第一個錯誤的版本。 呼叫 isBadVersion(3) -> false 呼叫 isBadVersion(5) -> true 呼叫 isBadVersion(4) -> true 所以,4 是第一個錯誤的版本。
這個題一看馬上就想到了二分法,我們要尋找第一個錯誤版本,也就是要保留最後一個false之後的第一個true,所以在更新邊界的時候,右邊界就不用減1了,這樣最後當左右相等時一定是第一個true。另外如果中間值直接用左右相加除2的話可能會在大數時溢位,所以可以採用左邊界加移動項,移動項就是右邊界減左邊界再除2。
C++原始碼:
// Forward declaration of isBadVersion API.
bool isBadVersion(int version);
class Solution {
public:
int firstBadVersion(int n) {
int left = 1;
int right = n;
while(left<right)
{
int mid = left+(right-left)/2;
if (isBadVersion(mid)==true)
right = mid;
else
left = mid+1;
}
return left;
}
};
python3原始碼:
# The isBadVersion API is already defined for you.
# @param version, an integer
# @return a bool
# def isBadVersion(version):
class Solution:
def firstBadVersion(self, n):
"""
:type n: int
:rtype: int
"""
left = 1
right = n
while left<right:
mid = left + (right-left)//2
if isBadVersion(mid):
right = mid
else:
left = mid+1
return left