常犯的问题
如果考虑不清楚边界问题就会导致写代码模模糊糊,似懂非懂,每次解题就碰运气,完整默写出二分模板,对了就过,不对就换一个,总是纠结不清。
二分查找本是高效的算法,若每次都因为边界问题而导致解题的不顺畅,这显然是致命的。
以下给出一个非常通用的方法,可以避免以往的所有坑点,以后再也不担心出现死循环了hhh,篇幅不长,大家耐心往下看。
注:建议大家先了解二分查找的基本概念再往下阅读,不少文章用大量图片甚至动图来解释清楚了。
图解二分法
这里我们分成两种情况讨论,每一种情况对应一个模板,模板在后面会给到。
情况一:左区间不成立,右区间成立
注:ans是所求的解的下标,check是检查函数。始终明确一点,闭区间[l, r]一定是包含了ans的。
- 当mid指向左区间时,check(mid)为false,说明解必定落在右侧区间,并且在mid上是不满足条件的,所以我们把区间更新为[mid + 1, r]。
- 当mid指向右区间时,check(mid)为true,说明解必定落在左侧区间,此时在mid上是有可能满足条件的(mid可能会刚好落在ans上),所以我们把区间更新为[l, mid]
情况二:左区间成立,右区间不成立
分析过程与情况一类似。
注:ans是所求的解的下标,check是检查函数。始终明确一点,闭区间[l, r]一定是包含了ans的。
- 当mid指向左区间时,check(mid)为true,说明解必定落在右侧区间,并且在mid上是可能满足条件的(因为mid可能会刚好落在ans上),所以我们把区间更新为[mid, r]。
- 当mid指向右区间时,check(mid)为false,说明解必定落在左侧区间,此时在mid上是不满足条件的,所以我们把区间更新为[l, mid - 1]。
模板代码
这两个模板分别对应不同的情况,要视情况而用。
情况一:左区间不成立,右区间成立
void bSearch(int l, int r){
while(l < r){
int mid = l + r >> 1;
if(check(mid)) r = mid;
else l = mid + 1;
}
return l;
}
情况二:左区间成立,右区间不成立
void bSearch_2(int l, int r){
while(l < r){
// 加1的是因为当r - l等于1时,mid始终为l,也就是check(mid)永为true,造成死循环。
int mid = l + r + 1 >> 1;
if(check(mid)) l = mid;
else r = mid - 1;
}
}
一个简单例子的应用
题目:给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
示例 1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
示例 2:
>输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
代码实现:
class Solution {
public:
int search(vector<int>& nums, int target) {
int l = 0, r = nums.size() - 1;
while(l < r){
int mid = l + r >> 1;
if(nums[mid] >= target) r = mid;
else l = mid + 1;
}
if(nums[l] == target) return l;
else return -1;
}
};