题目:
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 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
思路:
题目中要求,尽量减少对API的调用,其目的是将解决问题的该算法的性能尽可能优化,因此我们采用二分查找的方法来优化解决方法。在常规二分查找的格式下需要根据实际要求进行一些小的改动,因为在出现第一个次品之前产品均正常,出现第一个次品之后产品均为次品,因此在比较过程中,若isBadVersion(mid)为假,说明第一个次品一定在mid右边,故low = mid+1;
若isBadVersion(mid)为真,说明mid处一定为次品,但是不是第一个次品无法判定,需要对low指针继续调整缩小范围,直到最后跳出while循环的条件(low==high),此时low = high = mid,且该位置即为第一个次品出现的位置。
关于二分查找的代码实现可见我的另一篇博文:算法——二分查找
代码如下:
/* The isBadVersion API is defined in the parent class VersionControl.
boolean isBadVersion(int version); */
public class Solution extends VersionControl {
public int firstBadVersion(int n) {
int low = 1;
int high = n;
while(low < high){
int mid = low + (high - low)/2;//防止数据溢出
if(!isBadVersion(mid)) low = mid + 1;
else if(isBadVersion(mid)) high = mid;
}
return low;
}
}