问题背景
「一本通1.2 练习1」
题目描述
对于给定的一个长度为 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 个空格隔开的非负整数 Ai,含义如题目所述。
输出格式
仅包含一个正整数,即每段和最大值最小为多少。
样例输入1
5 3
4 2 4 5 1
样例输出1
6
注释说明
对于 20%的数据,有 N≤10;
对于 40% 的数据,有 N≤1000;
对于 100的数据,有 N≤10^5,M≤N, Ai 之和不超过 10^9。
一眼二分
#include<bits/stdc++.h>
using namespace std;
int n,m,a[100002];
bool check(int x){
int c=0,h=0;
for(int i=1;i<=n;i++){
if(h+a[i]>x){
if(a[i]>x)return 0;
h=a[i];
c++;
}
else h+=a[i];
}
if(h!=0)c++;
if(c>m)return 0;
return 1;
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
int l=1,r=1e9,mid;
while(l<=r){
mid=(l+r)/2;
//printf("");
if(check(mid))r=mid-1;
else l=mid+1;
}
printf("%d\n",l);
// cout<<r<<"\n";
}