整数二分
整数二分模板一共有两个,分别适用于不同情况。
算法思路:假设目标值在闭区间[l, r]中, 每次将区间长度缩小一半,当l = r时,我们就找到了目标值。
版本1通常适用于满足某个条件的最小值(mid = l + r >> 1),版本2通常适用于满足某个条件的最大值(mid = l + r + 1 >> 1)。
版本1
当我们将区间 [l, r]
划分成 [l, mid]
和 [mid + 1, r]
时,其更新操作是 r = mid
或者 l = mid + 1
;,计算 mid
时不需要加1。
C++ 代码模板:
int bsearch_1(int l, int r)
{
while (l < r)
{
int mid = l + r >> 1;
if (check(mid)) r = mid;
else l = mid + 1;
}
return l;
}
版本2
当我们将区间 [l, r]
划分成 [l, mid - 1]
和 [mid, r]
时,其更新操作是 r = mid - 1
或者 l = mid
;此时为了防止死循环,计算 mid
时需要加1。
C++ 代码模板:
int bsearch_2(int l, int r)
{
while (l < r)
{
int mid = l + r + 1 >> 1;
if (check(mid)) l = mid;
else r = mid - 1;
}
return l;
}
版本3
我自己常用的模板,以下代码作用同版本2。
C++ 代码模板:
int bsearch_3(int l, int r)
{
int ans;
while (l <= r) {
int mid = l + r >> 1;
if (check(mid)) {
ans = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
return ans;
}
浮点数二分
如果保留n位小数,while条件写 r - l > 1e-(n+2)
。
C++ 代码模板:
double bsearch_4(double l, double r)
{
while (r - l > 1e-6) //保留4位小数
{
double mid = (l + r) / 2;
if (check(mid)) l = mid;
else r = mid;
}
return l;
}