Leetcode 278.第一个错误的版本

Time: 20190909
Type: Easy

题目描述

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

假设你有 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

仍然采用相邻时退出的策略,最后在循环之外做一下判断,二分搜索只做范围缩小。

本题是典型的xxxxxxxxxxxxxoooooooo类的问题,每次搜索丢一半。

代码

# 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
        """
        # 版本天然有序,所以二分搜索,寻找第一个出错的版本
     
        start, end = 1, n 
        # 二分搜索只负责缩小范围, start + 1 == n时退出
        while start + 1 < end:
            
            mid = start + (end - start) // 2
            print(mid)
            if isBadVersion(mid):
                # 向前走
                end = mid
            else:
                start = mid
        
        # print(isBadVersion(1))
        if isBadVersion(start):
            return start
        if isBadVersion(end):
            return end

要注意最后的判断,先判断左边的,再判断右边的,因为最后一定会收缩到end指向错误的,low指向错误的,或者非错误的。

END.

你可能感兴趣的:(LeetCode)