「1.2」数列分段II

 

问题背景

「一本通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";
}