题目描述
对于给定的一个长度为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。
输入输出格式
输入格式:
输入文件divide_b.in的第1行包含两个正整数N,M,第2行包含N个空格隔开的非负整数A[i],含义如题目所述。
输出格式:
输出文件divide_b.out仅包含一个正整数,即每段和最大值最小为多少。
输入输出样例
输入样例#1:
5 3
4 2 4 5 1
输出样例#1:
6
说明
对于20%的数据,有N≤10;
对于40%的数据,有N≤1000;
对于100%的数据,有N≤100000,M≤N, A[i]之和不超过10^9。
Q
虽然很水,但是调了好长时间;
注意:
1. 二分边界l=max(a[i]),r=sum(a[i]);
如果l=0 or 1;
对于 :
5 2
1 2 3 7897979 4;
答案不会是7897979!!!!!
卡了我半年!!!!!
2. 还有一个小细节,check时,可能最后剩下几个数加起来不到x,但组数也要加,所以加一个特判;
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,l=1,r,a[200001];
bool check(int x)
{
int tot=0,sum=0;
for(int i=1;i<=n;i++)
{
if(sum+a[i]>=x) tot++,sum=0;
sum+=a[i];
}
if(sum) tot++;//判断是否有多余的数没统计;
if(tot>m) return true;//x不够大,增大x;
else return false;//tot=m时,由于让max尽量小,所以尽量减小x;
}
void solve()
{
cin>>n>>m;
for(int i=1;i<=n;i++) scanf("%d",&a[i]),r+=a[i],l=max(l,a[i]);
while(r-l>1)
{
int mid=(l+r)>>1;
if(check(mid)) l=mid;
else r=mid;
}
cout<<l;
}
int main()
{
solve();
return 0;
}