leetCode-278. 第一个错误的版本

807 阅读1分钟

题目链接

题目描述

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

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

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

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

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

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

题解

思路:二分查找

根据题目错误的版本之后的所有版本都是错的可倒推到,若一个版本是正确的,这个版本之前的缩影版本都是正确的。可以利用这个进行二分查找。

我们可以将左右边界分为设为1和n,然后根据左右边界找到中间的版本mid,判断mid版本是否正确,若正确,则说明mid以及之前的版本都正确,将左边界移动到mid+1,若错误,则说明mid以及之后的版本都错误,将右边界移动到mid,直到左边界不小于右边界为止,这个左边界就是第一个错误的版本。

代码

func firstBadVersion(_ n: Int) -> Int {
    var left = 1
    var right = n
    while left < right {
        let mid = (left + right) / 2
        if isBadVersion(mid) {
            right = mid
        } else {
            left = mid + 1
        }
    }
    return left
}