基于JavaScript实现查找第一个错误的版本(二分查找算法)力扣(LeetCode)
题目
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/first-bad-version
算法解析
-
由于应尽量减少调用API接口,故利用二分查找提高查找效率。
-
定义一个start和end,用以表示查找的区域,mid表示中间值。
-
进行二分循环查找,如果isBadVersion(mid)是true,则再进行判断前面一个版本是否出错,即isBadVersion(mid-1)?
- 如果为false,就表示mid是第一个出错的版本,返回结果mid;
- 如果为true,则前面仍有出错的版本,就把mid-1的值赋给end,再次二分,改变查找区域,循环查找
-
如果isBadVersion(mid)是false,则表示mid以及之前的版本是正确的,就把mid+1的值给start,改变查找区域,进行循环查找。
代码实现
var solution = function(isBadVersion) {
/**
* @param {integer} n Total versions
* @return {integer} The first bad version
*/
return function(n) {
var start = 1;
var end = n;
while(start <= end) {
var mid = start + Math.floor((end - start) / 2);
if(isBadVersion(mid) == true) {
if(mid == 1 || isBadVersion(mid - 1) == false){
return mid;
}else if(isBadVersion(mid - 1) == true){
end = mid - 1;
}
}else if(isBadVersion(mid) == false) {
start = mid + 1;
}
}
return 0;
};
}