LibreOj #10014. 「一本通 1.2 练习 1」数列分段 II

经典二分稍作更改

#include<bits/stdc++.h>
using namespace std;
int a[100005]={0};
int n,m;
bool check(int x)//判断最大值为x时能否分成m段
{
    int num=1,sum=0;
    for(int i=0;i<n;i++)
    {
        if(sum+a[i]>x)
        {
            sum=a[i];
            num++;
        }
        else
        sum+=a[i];
    }
    if(num<=m)
    {
        return true;
    }
    else return false;
}
int main()
{
    scanf("%d%d",&n,&m);
    int max,sum;
    scanf("%d",&a[0]);
    sum=a[0],max=a[0];
    for(int i=1;i<n;i++)
    {
        scanf("%d",&a[i]);
        sum+=a[i];
        if(a[i]>max)max=a[i];
    }
    int mid,ans,l=max,r=sum;//l（单个数值的最大值）最小，r（总和）最大
    while(l<=r)
    {
        mid=(l+r)/2;
        if(check(mid)==true)
        {
            ans=mid;
            r=mid-1;//减小上限
        }
        else
        l=mid+1;//增加下限
    }
	printf("%d\n",ans);
	return 0;
}