你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
示例:
给定 n = 5,并且 version = 4 是第一个错误的版本。
调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true
所以,4 是第一个错误的版本。
思路:典型的二分查找
因为版本是从1开始的,所以我们在这个题中令初始状态low=1、high=n。
找到mid之后,判断mid是否为错误版本,如果不是错误版本,说明第一个错误版本在mid的后边,更新low=mid+1;如果是错误版本,说明第一个错误版本有可能是mid也有可能在mid的前面,进一步判定,如果mid-1是正确版本,说明mid就是第一个错误版本,返回mid,否则说明第一个错误版本还在前面,更新high=mid-1.
# The isBadVersion API is already defined for you.
# @param version, an integer
# @return a bool
# def isBadVersion(version):
class Solution(object):
def firstBadVersion(self, n):
"""
:type n: int
:rtype: int
"""
low = 1
high = n
while low<=high:
mid = (low+high)//2
if isBadVersion(mid) == True:
if isBadVersion(mid-1) == False:
return mid
else:
high = mid-1
else:
low = mid+1