数列分段 Section II

题目描述

对于给定的一个长度为N的正整数数列 A_{1\sim N}A1∼N，现要将其分成 MM（M\leq NM≤N）段，并要求每段连续，且每段和的最大值最小。

关于最大值最小：

例如一数列 4\ 2\ 4\ 5\ 14 2 4 5 1 要分成 33 段。

将其如下分段：

[4\ 2][4\ 5][1][4 2][4 5][1]

第一段和为 66，第 22 段和为 99，第 33 段和为 11，和最大值为 99。

将其如下分段：

[4][2\ 4][5\ 1][4][2 4][5 1]

第一段和为 44，第 22 段和为 66，第 33 段和为 66，和最大值为 66。

并且无论如何分段，最大值不会小于 66。

所以可以得到要将数列 4\ 2\ 4\ 5\ 14 2 4 5 1 要分成 33 段，每段和的最大值最小为 66。

输入格式

第 11 行包含两个正整数 N,MN,M。

第 22 行包含 NN 个空格隔开的非负整数 A_iAi，含义如题目所述。

输出格式

一个正整数，即每段和最大值最小为多少。

输入输出样例

输入 #1

5 3
4 2 4 5 1

输出 #1

6

分析

上下有界且单调，最大值求最小，最小值求最大，想到二分搜索的应用

while (left <= right)
    {
        mid = (left + right) / 2;
        if (check(mid))
            left = mid + 1;
        else
            right = mid - 1;
    }

基本代码如上，现在对check写函数，求每段和的最大值，则，每次规定mid,对，遍历数据，只要不超过mid就把它们放在一个段内，求当前选的这个最大值的最小（mid），求每段不大于mid的最小划分数，，看看能分多少段。r 设为最大的数，即可以设为数列的和，l 设为较小的数可以设为数列的最大值

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100004;
int n, m;
int arr[N];
int check(int k)
{
    int i, sum , ans;
    sum = 0;
    ans = 1;
    for (i = 0; i < n; i++)
    {
        if (sum + arr[i] <= k)
        {
            sum += arr[i];
        }
        else
        {
            ans++;
            sum = arr[i];
        }
    }
    return ans > m;//如果分段大于要分的段数了，说明mid选大了，反之则选小了
}
int main(void)
{
    cin >> n >> m;
    int left=0, right=0, mid;
    for (int i = 0; i < n; i++)
    {
        cin >> arr[i];
        right += arr[i];
        left = max(left, arr[i]);
    }
    while (left <= right)
    {
        mid = (left + right) / 2;
        if (check(mid))
            left = mid + 1;
        else
            right = mid - 1;
    }
    cout << left;
}