first_bad_version.py

"""
278. 第一个错误的版本
二分查找 交互
简单


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

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

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

 
示例 1：

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

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

提示：

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


来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/first-bad-version
"""

bad = -1


# The isBadVersion API is already defined for you.
# @param version, an integer
# @return an integer
def isBadVersion(version: int) -> bool:
    print('isBadVersion({}) -> {}'.format(version, version >= bad))
    return version >= bad


class Solution:
    def firstBadVersion(self, n):
        """
        :type n: int
        :rtype: int
        """
        return self.binary_search(1, n)

    def binary_search(self, left, right):
        """
        :param left: int
        :param right: int
        :return: int
        """
        if left >= right:
            if isBadVersion(right):
                return right
            else:
                return left + 1
        middle = (left + right + 1) // 2
        if isBadVersion(middle):
            return self.binary_search(left, middle-1)
        else:
            return self.binary_search(middle, right)


if __name__ == '__main__':
    solution = Solution()

    n = 5
    bad = 4
    result = solution.firstBadVersion(n)
    print(result)
    assert result == 4

    n = 1
    bad = 1
    result = solution.firstBadVersion(n)
    print(result)
    assert result == 1

    n = 2
    bad = 2
    result = solution.firstBadVersion(n)
    print(result)
    assert result == 2

    n = 4
    bad = 3
    result = solution.firstBadVersion(n)
    print(result)
    assert result == 3