描述
对于给定的一个长度为 N 的正整数数列 A ,现要将其分成 M 段,并要求每段连续,且每段和的最大值最小。
例如,将数列 4 2 4 5 1 要分成 3 段:
若分为 [4 2][4 5][1],各段的和分别为 6, 9, 1 ,和的最大值为 9;
若分为 [4][2 4][5 1],各段的和分别为 4, 6, 6 ,和的最大值为 6;
并且无论如何分段,最大值不会小于 6。
所以可以得到要将数列 4 2 4 5 1 要分成 3 段,每段和的最大值最小为 6 。
输入
第1行包含两个正整数 N ,M;
第2行包含 N 个空格隔开的非负整数A[i],含义如题目所述。
输出
仅包含一个正整数,即每段和最大值最小为多少。
输入样例 1
5 3
4 2 4 5 1
输出样例 1
6
提示
二分,数组用十万
#include<iostream>
#include<algorithm>
using namespace std;
int a[1000100];
int n,m,nmax;
int sum=0;
bool judge(int mid)
{
int nsum=0;
int ans=1;
for(int i=1;i<=n;i++)
{
if(nsum+a[i]<=mid)
{
nsum+=a[i];
}
else
{
nsum=a[i];
ans++;
}
}
if(ans>m)
return true;
else
return false;
}
void work()
{
int mid;
int low=nmax,high=sum;
mid=(low+high)/2;
while(low<=high)
{
if(judge(mid)==true)
low=mid+1;
else
high=mid-1;
mid=(low+high)/2;
}
cout<<mid+1;
}
void init()
{
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum+=a[i];
nmax=max(nmax,a[i]);
}
}
int main()
{
cin>>n>>m;
init();
work();
return 0;
}