二分查找及二分答案

大家好，我又来讲东西了，关于二分，我做出以下讲解：

二分是什么？

二分就是在一定范围内的数组中查找符合条件的数（即一分为二的方法），意思很简单：

比如就是我们小时候都玩过的猜数字游戏其实没玩过，幼儿园老师让你们猜数{ 如果老师心里面想的数是 1 }，范围是（1 - 100），那么聪明的你肯定会想折半查找，先猜（100 + 1）/ 2 = 50，然后老师说大了，然后你就往小的猜就是（50 + 1）/ 2 = 25，老师又说大了，然后你又往小的猜（25 + 1）/ 2 = 13……最后经过了一段时间就得出了老师心中的那个数值 1

整体思路

        二分查找

像上面那样折半查找的方式可以看成这么一串式子：

当区间内只有一个元素时停止。

设开始时数组中元素个数为n。

第一次二分后:答案所在区间长度缩小一半:n / 2

第二次二分后:答案所在区间长度缩小一半:n / 4

第二次二分后:答案所在区间长度缩小一半:n / 8

……

很容易让人看出来如图:

对于一个判断条件：只剩下一个元素时（就是当左边的所查找的下标等于右边的查找下标时），我们令这两个指针为 l 与 r，这就是二分查找

二分答案

类似的，二分答案就是把所有的答案总结起来，在答案中查找为了方便，你还可以自定义一个函数来求答案

整体代码如下：

#include <iostream>
#include <vector>
using namespace std;
// 快速读取整数函数
int read() {
    int x = 0, f = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-') f = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        x = x * 10 + c - '0';
        c = getchar();
    }
    return x * f;
}
bool check(const vector<long long>& a, long long n, long long m, long long h) {
    long long group_count = 1; // 当前分组数量
    long long group_sum = 0;   // 当前分组的和
    for (int i = 0; i < n; ++i) {
        group_sum += a[i];
        if (group_sum > h) {
            group_count++;
            group_sum = a[i];
        }
    }
    return group_count <= m;
}
int main() {
    long long n = read();
    long long m = read();
    vector<long long> a(n);
    long long right = 0; // 二分右边界：数组总和
    long long left = 0;  // 二分左边界：数组中的最大值
    for (int i = 0; i < n; ++i) {
        a[i] = read();
        right += a[i];
        if (a[i] > left) {
            left = a[i];
        }
    }
    long long ans = 0;
    // 二分查找循环：在[left, right]区间内查找最小的最大子数组和
    while (left <= right) {
        // 计算中间值，使用left + (right - left) / 2的写法避免溢出
        // 例如：当left和right都是接近LLONG_MAX的数时，(left + right)会溢出
        long long mid = left + (right - left) / 2;
        
        // 检查是否可以将数组分割成不超过m个连续子数组，每个子数组的和不超过mid
        if (check(a, n, m, mid)) {
            // 如果可以分割，说明mid是一个可行解，我们尝试寻找更小的可行解
            ans = mid;         // 记录当前可行解
            right = mid - 1;   // 缩小搜索范围，在左半部分继续查找
        } else {
            // 如果不可以分割，说明mid太小，需要增大mid的值
            left = mid + 1;    // 缩小搜索范围，在右半部分继续查找
        }
    }
    // 输出最终结果：最小的最大子数组和
    cout << ans << endl;
    return 0;
}

这个是洛谷P1182 数列分段 Section II   https://www.luogu.com.cn/problem/P1182