问题描述

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

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

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

示例:

给定 n = 5,并且 version = 4 是第一个错误的版本。

调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true

所以,4 是第一个错误的版本。

解决方案

解题思路:

  • 如果猜的版本为bad,那么右边界替换成当前位置
  • 如果猜的版本不为bad,那么左边界为当前值+1
  • 到最后一定是左边界为好的版本,右边界为Bad版本
  • 最后返回右边界

时间复杂度:O(logn)

class Solution:
    def firstBadVersion(self, n):
        low = 1
        high = n
        while low < high:
            mid = (low + high) //2
            if isBadVersion(mid) is True:
                high = mid
            else:
                low = mid + 1
        return high

本博客所有文章除特别声明外,均采用 CC BY-SA 3.0协议 。转载请注明出处!

狄克斯特拉算法 上一篇
【Linux编程】进程间通信(IPC) 下一篇