题目描述:
对于给定的一个长度为N的正整数数列A−i,现要将其分成M(M≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数列42451要分成3段
将其如下分段:
[4 2][4 5][1]
第一段和为6,第2段和为9,第3段和为1,和最大值为9。
将其如下分段:
[4][2 4][5 1]
第一段和为4,第2段和为6,第3段和为6,和最大值为6。
并且无论如何分段,最大值不会小于6。
所以可以得到要将数列42451要分成3段,每段和的最大值最小为6。
输入输出格式
输入格式:
第1行包含两个正整数N,M。
第2行包含N个空格隔开的非负整数A_i,含义如题目所述。
这里是引用
输出格式:
一个正整数,即每段和最大值最小为多少。
思路:
最大值的最小化,显然用二分来划定区间,个人感觉二分最难的就是自定义judge函数后面的l和r与mid的关系,有时要偏移一位,有时又不用。这里l和r都需要偏移一位,当不满足时,l=mid+1没有问题,但是满足的时候,有可能这个mid就是要求的,而这里却让r=mid-1,如果比mid小的都不满足呢?不要急,它会在一次次缩小区间后,让l变回刚开始的mid了。
代码:
#include <bits/stdc++.h>
using namespace std;
int n, m, a[100005];
int m_max;
int l=0, mid, r;
bool judge(int x)
{
int sum=1, t=0;
for (int i=1; i<=n; i++){
if (t+a[i]<=x){
t+=a[i];
}else{
t=a[i];
sum++;
}
}
if (sum<=m) return true;
else return false;
}
void func()
{
while (l<=r){
mid=(l+r)/2;
if (judge(mid))
r=mid-1;
else
l=mid+1;
}
m_max=l;
}
int main()
{
cin>>n>>m;
for (int i=1; i<=n; i++){
cin>>a[i];
r+=a[i];
l=l>a[i]?l:a[i];
}
func();
cout<<m_max;
return 0;
}