1436：数列分段II

1436：数列分段II

题解

二分答案

我们最终答案的取值区间是[  max(a[i])  ,   ∑a[i]  ]

设定 l=max(a[i]) , r=∑a[i]  , mid不断二分

mid表示每段和的最大值，也就是每段和都不超过mid

放到check函数里，计算一下在mid为最大值的情况下可以分成多少段

如果段数 cnt > m ，说明这个mid小了，它还可以再大一点

如果段数 cnt <= m , 说明这个mid大了，那么它就要小一点了，由于此时cnt可能等于m,这个mid为候选答案，记录下来（如果他是真正答案，最后输出的就是他，否则他会被更新为一个更小的）

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
#include<functional>

using namespace std;

int n,m,ans,l=,r=;
int a[];
int sum[]; 

int check(int x)
{
    int cnt=,now=;
    for(int i=;i<=n;i++)
    {
        if(now+a[i]<=x) now+=a[i];
        else
        {
            cnt++;
            now=a[i];
        }
    }
    return cnt<=m;
}

int main()
{
     scanf("%d%d",&n,&m);
     for(int i=;i<=n;i++)
     {
         scanf("%d",&a[i]);
         l=max(l,a[i]);
         r+=a[i];
     }

     while(l<=r)
     {
         int mid=(r+l)/;
         if(check(mid))
         {
             ans=mid;
             r=mid-;
        }
        else l=mid+;
     }

    printf("%d",ans);

}