刷题：leetcode 278二叉搜索 变体

题目描述

你是产品经理，目前正在带领一个团队开发新的产品。不幸的是，你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的，所以错误的版本之后的所有版本都是错的。

假设你有 n 个版本 [1, 2, …, n]，你想找出导致之后所有版本出错的第一个错误的版本。

你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。

输入：n = 5, bad = 4
输出：4
解释：
调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true
所以，4 是第一个错误的版本。

输入：n = 1, bad = 1
输出：1

1 <= bad <= n <= 231 - 1

分析

假设：
0 表示正常；
1 表示错误的版本；
真实是输入为：
0 0 0 0 0 0 1 1 1 1 1 1 1

相对于普通的二叉搜索，只有一个target。该变体有多个target。因此变体如下：
mid 为中间值；
mid_2 为 mid的下一个版本；
若 mid 正常，mid_2 错误，则 mid_2 即为返回值。

二叉搜索，循环实现

# 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
        """
        if isBadVersion(1):
            return 1
        left = 1
        right = n
        while(left <= right):
            mid = int((left +right) /2)
            mid_2 = mid + 1
            
            if isBadVersion(mid_2) == False:
                left = mid_2
                right = right
            elif isBadVersion(mid) == True:
                left = left
                right = mid
            else:
                return mid_2