题目描述
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n
个版本 [1, 2, ..., n]
,你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用bool isBadVersion(version)
接口来判断版本号 version
是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
示例
给定 n = 5,并且 version = 4 是第一个错误的版本。
调用 isBadVersion(3) -> false
调用 isBadVersion(5) -> true
调用 isBadVersion(4) -> true
所以,4 是第一个错误的版本。
题解
二分查找
解题关键:调用isBadVersion()
,返回True
时,说明目标元素一定小于或等于当前值,则调整右边界为当前元素的索引;返回False
时,说明目标元素一定大于当前值,则调整左边界为当前元素右边的索引。终止条件为l=r
,即找到了目标元素。
例如:
[1,2,3,4,5], 从2开始为Badversion
step1. l =1, r = 5, mid = 3
step2. isBadVersion(3) -> True, 则r = mid = 3
step3. mid = (l+r)//2 = 2, isBadVersion(2) -> True, 则r = mid = 2
step4. mid = (l+r)//2 = 1, isBadVersion(1) -> False, 则l = mid+1 = 2
step5. l == r,循环结束,目标元素为l = 2
完整的代码实现:
class Solution {
public:
int firstBadVersion(int n)
{
int l = 1;
int r = n;
int mid;
while (l < r)
{
mid = l + (r - l) / 2;
if (isBadVersion(mid))
r = mid;
else
l = mid +1;
}
return l;
}
};
- 时间复杂度: O ( l o g N ) O(log N) O(logN),二分查找的复杂度为 l o g N log N logN。
- 空间复杂度: O ( 1 ) O(1) O(1)