二分【C++|AcWing|算法】

$二分$

一、整数二分

整数二分有两个模板，模板1会找到最左边的二分点，模板2会找到最右边的二分点。

模板1：尽量往左找答案；mid 不用加一。
模板2：尽量往右找答案；mid 要加一。

模板1：

int binary_search(int l, int r, int a[])
{
    while (l < r)
    {
        int mid = l + r >> 1;
        if (check(a[mid]))  // check是否满足右半区规则
            r = mid;
        else
            l = mid + 1;
    }
    return l;
}

模板2：

int binary_search(int l, int r, int a[])
{
    while (l < r)
    {
        int mid = l + r + 1 >> 1;   // 特别注意，模板2中mid需要先加1再除2
        if (check(a[mid]))  // check是否满足左半区规则
            l = mid;
        else
            r = mid - 1;
    }
    return l;
}

AcWing_789 数的范围

给定一个按照升序排列的长度为 $n$ 的整数数组，以及 $q$ 个查询。

对于每个查询，返回一个元素 $k$ 的起始位置和终止位置（位置从 $0 $ 开始计数）。

如果数组中不存在该元素，则返回 -1 -1。

输入格式

第一行包含整数 $n$ 和 $q$，表示数组长度和询问个数。

第二行包含 $n$ 个整数（均在 $1\sim 10000$ 范围内），表示完整数组。

接下来 $q$ 行，每行包含一个整数 $k$，表示一个询问元素。

输出格式

共 $q$ 行，每行包含两个整数，表示所求元素的起始位置和终止位置。

如果数组中不存在该元素，则返回 -1 -1。

数据范围

$1\le n\le 100000$
$1\le q\le 10000$
$1\le k\le 10000$

输入样例：

6 3
1 2 2 3 3 4
3
4
5

输出样例：

3 4
5 5
-1 -1

解题：

这题其实是模板1和模板2的综合，只要求出最左边的二分点和最右边的二分点即可。

模板1找答案时会尽量往左边找，就会找到该数的起始位置；而模板2找答案时会尽往右边找，就会找到该数的终止位置。分别使用模板1和模板2进行一次二分就可以了。

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int a[N];

int main()
{
    int n, q;
    cin >> n >> q;
    for (int i = 0; i < n; i++)
        cin >> a[i];

    while (q--)
    {
        int x;
        cin >> x;

        int l = 0, r = n - 1;
        while (l < r)
        {//模板1
            int mid = l + r >> 1;
            if (a[mid] >= x)
                r = mid;
            else
                l = mid + 1;
        }
        if (a[l] != x)
        {
            cout << "-1 -1" << endl;
            continue;
        }
        cout << l << " ";

        l = 0, r = n - 1;
        while (l < r)
        {//模板2
            int mid = l + r + 1 >> 1;
            if (a[mid] <= x)
                l = mid;
            else
                r = mid - 1;
        }
        cout << l << endl;
    }

    return 0;
}

二、浮点数二分

浮点数的二分非常简单，因为不需要考虑边界。下面直接看一道题。

AcWing_790 数的三次方根

给定一个浮点数 $n$，求它的三次方根。

输入格式

共一行，包含一个浮点数 $n$。

输出格式

共一行，包含一个浮点数，表示问题的解。

注意，结果保留 $6$ 位小数。

数据范围

$-10000\le n\le 10000$

输入样例：

1000.00

输出样例：

10.000000

解题：

无需考虑边界，直接二分。

#include <iostream>
#include <iomanip>
using namespace std;
int main()
{
    double x;
    cin >> x;

    double l = -100, r = 100;
    while (r - l > 1e-8)
    {
        double mid = (l + r) / 2;
        if (mid * mid * mid >= x)   //这里判断小于等于当然也是可以的，下面也改一下就行了
            r = mid;
        else
            l = mid;
    }
    cout << fixed << setprecision(6) << l << endl;

    return 0;
}

$易错：代码中的l和r不可以分别取-x和x$
因为当 $x$ 为负数时，就会出现 $l$ 初始值就大于 $r$。

l = -fabs(x), r = fabs(x);

这样也是不可行的，当 $x$ 绝对值小于 $1$ 时，$l和r$ 取得太小，永远也二分不到答案。

所以最好的方法是直接设置两个合理的初始值。