题目描述
对于给定的一个长度为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仅包含一个正整数,即每段和最大值最小为多少。
输入:5 3
4 2 4 5 1
输出:
6
用二分枚举边界 要判断特殊情况 比如 5 3 1 1 1 1 9999 这种情况
#include<bits/stdc++.h>
using namespace std;
#define maxn 100000+100
long long a[maxn];
int main(){
int n,m;
cin>>n>>m;
memset(a,0,sizeof(a));
int s=0;
int mx=0;
for(int j=1;j<=n;j++){
int x;
scanf("%d",&x);
mx=max(mx,x);
a[j]=a[j-1]+x;
}
s=a[n];
int l=0,r=s;
int mid;
while(l<=r){
mid=(l+r)/2;
int i=1;
int ans=1;
for(int j=1;j<=n;j++){
if(a[j]-a[i-1]>mid){
ans++;
i=j;
}
}
//cout<<l<<" "<<r<<" "<<ans<<endl;
if(ans>m){
l=mid+1;
}
else{
r=mid-1;
}
}
cout<<max(l,mx)<<endl;
return 0;
}