1436：数列分段II

1436：数列分段II

时间限制: 1000 ms 内存限制: 65536 KB
提交数: 1519 通过数: 693
【题目描述】
对于给定的一个长度为N的正整数数列A[i]，现要将其分成M（M≤N）段，并要求每段连续，且每段和的最大值最小。

关于最大值最小：

例如一数列4 2 4 5 1要分成3段

将其如下分段：

[4 2][4 5][1]

第一段和为6，第2段和为9，第3段和为1，和最大值为9。

将其如下分段：

[4][2 4][5 1]

第一段和为4，第2段和为6，第3段和为6，和最大值为6。

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

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

【输入】
第1行包含两个正整数N，M，第2行包含N个空格隔开的非负整数A[i]，含义如题目所述。

【输出】
仅包含一个正整数，即每段和最大值最小为多少。

【输入样例】
5 3
4 2 4 5 1
【输出样例】
6

#include <bits/stdc++.h>
#define ll long long
#define PII pair<int,int>
#define inf 1<<30

using namespace std;
const int N=1e6+5;
int n,m,a[N];

int f(int x)
{
    int num=0,s=0;
    for(int i=0;i<n;i++)
    {
        s+=a[i];
        if(s>x)
        {
            num++;
            s=a[i];
        }
    }
    num++;
    return num<=m;//够用
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m;
    int l=0,r=0,mid,res;
    for(int i=0;i<n;i++)
        cin >> a[i],l=max(l,a[i]),r+=a[i];
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(f(mid))
            r=mid-1,res=mid;
        else
            l=mid+1;
    }
    cout << res << endl;
    return 0;
}