算法(一) —— 二分模板
二分的模板有两套,主要的区别是根据判断时,是调整left还是right。
模板一:
其中:函数check(),表示判断mid的条件
//假设需要二分的区间是 [a,b]
int left = a, right = b;
while(left<right){
int mid = (left + right + 1)>>1;
//其实就是int mid = left + (right-left)/2 + 1;
if(check(mid)){
left = mid;
}else{
right = mid-1;
}
}
模板二:
//假设需要二分的区间是 [a,b]
int left = a, right = b;
while(left<right){
int mid = (left + right)>>1;
//其实就是int mid = left + (right-left)/2;
if(check(mid)){
right = mid;
}else{
left = mid+1;
}
}
总结:
模板一计算mid的时候是要加一的,具体原因是避免发生死循环,因为不论是 用>>1 还是用 /2 ,整数都是向下取整。
举个例子:left = 0, right = 1, 如果不加一的话,(left+right)>>1的结果还是mid = 0,就会陷入死循环。
还有一种可以避免死循环的方法,在最后,只会剩下两个数,正确的答案一定是两数的某一个(因为 left, mid, right 有两个是相等的),因此可以把while()循环的条件改成left!=mid 或right!=mid 就可以跳出来,最后只需要把两个数带入条件找到符合的即可。