对于二分查找我们可以有很多变形的题来做,那万变不离其宗的还是最原始的二分查找。在此基础上我们进行一些操作去完成更加复杂的题目。接下来给大家分享分享。
模板
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
class Solution {
public:
int search(vector<int>& nums, int target) {
int start = 0;
int end = nums.size()-1;
while(start<=end){
int middle = (start+end)/2;
if(nums[middle]==target)
return middle;
else if(nums[middle]<target)
start=middle+1;
else
end = middle-1;
}
return -1;
}
};
例题1
你是产品经理,目前正在带领一个团队开发新的产品。不幸的是,你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的,所以错误的版本之后的所有版本都是错的。
假设你有 n 个版本 [1, 2, …, n],你想找出导致之后所有版本出错的第一个错误的版本。
你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。
力扣(LeetCode)
链接:https://leetcode-cn.com/problems/first-bad-version
我们审题便不难发现它是一个二分查找的题,由此我们可以加上一些判断条件即可
代码如下
class Solution {
public:
int firstBadVersion(int n) {
int start = 1;
int end = n;
int middle;
while(start<=end){
middle =start+(end-start)/2;
if(isBadVersion(middle) == true && isBadVersion(middle-1) == false)
break;
else if( isBadVersion(middle) == true){
end = middle-1;
}else{
start = middle+1;
}
}
return middle;
}
};
例题2
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
我们要判断插入的位置,这块要注意,假设最终我们没有找到值,那如何插入进去是我们该思考的。
我们要注意此处的middle与target的关系,如果middle小于target,那应该插入后一位,middle>target,则应该插入middle的位置。当然如果等于则在前面已经找到了。
代码
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int start =0;
int end = nums.size()-1;
int mid;
if(nums[start] > target)
return start;
if( nums [end] < target)
return end+1;
while(start<=end){
mid = start + (end-start)/2;
if(nums[mid]==target)
return mid;
else if (nums[mid]<target)
start = mid+1;
else
end = mid-1;
}
if(nums[mid]<target)
return mid+1;
else
return mid;
}
};
总结
你会发现这两道变形题最基本的还是原始算法,所以掌握好最基础的模板,是比较重要的。这样我们在写类似的题目时,可以更快先人一步。