二分算法的思想是:有一个原始的区间,每次取中点进行判断,并删去区间的一半,直到区间的左右端点相等,此时区间的左右端点即为答案
二分算法常用来求解具有单调性的问题,比如:
①在原始区间中,左边一段满足条件,右边一段不满足条件,可以利用二分算法来求解满足条件的最大值
②在原始区间中,左边一段不满足条件,右边一段满足条件,可以利用二分算法来求解满足条件的最小值
可以计算出二分的时间复杂度为 O(logn),是一种很高效的算法
整数域上的二分
求解上限
即满足条件的最大值
while (l < r) {
int mid = (l + r + 1) / 2;
if (check(mid))
l = mid;
else
r = mid - 1;
}
1、l = mid 和 r = mid - 1
①当 mid 满足条件时,mid 可能是正确答案,所以令 l = mid
②当 mid 不满足条件时,mid 不可能是正确答案,所以令 r = mid - 1
2、mid = (l + r + 1) / 2
区间的变化有两种情况,[mid,r] 和 [l, mid-1]
①对于 [l, mid-1],mid-1 不可能等于 r,所以不会死循环
②对于 [mid,r],mid 可能等于 l,所以可能会死循环
为了避免②中的情况,也就是避免让 mid 等于 l,我们可以先对 l+r 加 1,这样 mid 就不会等于 l
求解下限
即满足条件的最小值
while (l < r) {
int mid = (l + r) / 2;
if (check(mid))
r = mid;
else
l = mid + 1;
}
1、r = mid 和 l = mid + 1
①当 mid 满足条件时,答案可能为 mid,所以 r = mid
②当 mid 不满足条件时,答案不可能为 mid,所以 l = mid + 1
2、mid = (l + r) / 2
区间的变化有两种情况,[l,mid] 和 [mid+1, r]
①对于 [l, mid],mid 不可能等于 r,所以不会死循环
②对于 [mid+1, r],mid+1 不可能等于 l,所以不会死循环
实数域上的二分
因为是double类型,相对于整数上的二分更加简单,无需考虑边界情况
eps表示精度值,通常需要比答案要求的精度再小两个数量级
while (l + eps < r) {
double mid = (l + r) / 2;
if (check(mid))
r = mid;
else
l = mid;
}
根据具体情况判断是改变 l 还是改变 r
STL库中封装好的函数
在STL库中,有封装好的二分算法函数,包含在 algorithm 头文件中
binary_search函数
binary_search(arr, arr + n, x):在(非递减的)数组中二分查找元素 x 是否出现
如果找到元素 x 则返回 true,如果找不到则返回 false。
binary_search(arr, arr + n, x);
lower_bound函数
lower_bound(arr, arr + n, x):在(非递减的)数组中二分查找第一个大于或等于 x 的数字
如果找到则返回该数字的地址,如果找不到则返回下标 n 的地址。
通过返回的地址减去数组首地址,就可以得到数字在数组中的下标。
int idx = lower_bound(arr, arr + n, x) - arr;
这个函数在离散化时经常用到
upper_bound函数
upper_bound(arr, arr + n, x):从(非递减的)数组二分查找第一个大于 x 的数字
如果找到则返回该数字的地址,如果找不到则返回下标 n 的地址。
通过返回的地址减去数组首地址,就可以得到数字在数组中的下标。
int idx = upper_bound(arr, arr + n, x) - arr;