题目描述
对于给定的一个长度为N的正整数数列A-iA−i,现要将其分成M(M≤N)M(M≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数列4 2 4 5 142451要分成33段
将其如下分段:
[4 2][4 5][1][42][45][1]
第一段和为66,第22段和为99,第33段和为11,和最大值为99。
将其如下分段:
[4][2 4][5 1][4][24][51]
第一段和为44,第22段和为66,第33段和为66,和最大值为66。
并且无论如何分段,最大值不会小于66。
所以可以得到要将数列4 2 4 5 142451要分成33段,每段和的最大值最小为66。
输入输出格式
输入格式:
第1行包含两个正整数N,M。
第2行包含NN个空格隔开的非负整数A_iAi,含义如题目所述。
输出格式:
一个正整数,即每段和最大值最小为多少。
输入输出样例
输入样例#1
5 3 4 2 4 5 1
输出样例#1
6
说明
对于20\%20%的数据,有N≤10N≤10;
对于40\%40%的数据,有N≤1000N≤1000;
对于100\%100%的数据,有N≤100000,M≤N, A_iN≤100000,M≤N,Ai之和不超过10^9109。
思路
因为是求最大的最小值,所以先考虑用二分答案。
这个二分的确难想,其实也可以避开前缀和,贪心一把。
#include <stdio.h>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m,s,a[100001],mid,r,l;
inline bool check(int x)//这个函数的本质是在a数组内能否每组的和比m小
{
int i,cnt(0),lxy(0);
for(i=1;i<=n;i++)//用贪心看最大值是否为x(即当前mid)
{
if(lxy+a[i]<=x)//这是个动态的过程
{
lxy+=a[i];
}
else
{
lxy=a[i];
cnt++;
}
}
return cnt>=m;//如果是,返回1;否则返回0
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
register int i,j,k;
cin>>n>>m;
for(i=1;i<=n;i++)
{
cin>>a[i];
l=max(l,a[i]);//l为a数组内的最大值,不能赋为0,因为题目中要求最大值最小,a数组的最大值一定是最小的(在无法分的时候)
r+=a[i];//r是区间和,同理,r一定保证区间最大
}
while(l<=r)
{
mid=(l+r)>>1;//取中间值
if(check(mid))
{
l=mid+1;
}
else
{
r=mid-1;
}
}
cout<<l<<endl;//由于l当前最小,所以输出l而不是mid或者r,不过这的确和一般的二分答案不一样。
return 0;
}