P1182 数列分段 Section II（二分）题解

原题地址：https://www.luogu.org/problem/P1182

题目描述

对于给定的一个长度为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。

输入输出格式

输入格式：

第1行包含两个正整数N，M。

第2行包含N个空格隔开的非负整数Ai ，含义如题目所述。

输出格式：

一个正整数，即每段和最大值最小为多少。

输入输出样例

输入样例#1：

5 3
4 2 4 5 1

输出样例#1：
6

说明

时空限制：1000ms 125M

对于20%的数据，有N≤10；

对于40%的数据，有N≤1000；

对于100%的数据，有N ≤ 100000，M ≤ N，Ai之和不超过10^9。

思路：
1、求数段和最大值中的最小的那个。那么肯定要在数列最大值与数列和之间查找，最初左边界为数列最大值，最初右边界为数列和，二分查找。

2、如果最少段数超过了m，即答案超过了mid，则要在mid右边继续寻找二分点；如果最少段数小于等于m，那么就要继续寻找左边的二分点。


代码如下：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std; 
const int N=1e6;
int n,m,a[N];
int l=0,r=0,mid;
bool judge(int t)
{
	int sum=0,num=0;
	for(int i=1;i<=n;i++)
	{
		//如果数列段和大于mid，则将a[i]重新赋给sum，同时段数+1 
		if(sum+a[i]>t)	
		{
			sum=a[i];
			num++; 
		} 
		else	//如果数列短和小于等于mid，则继续累加数列 
			sum+=a[i];
	}
	return  num<m;
}
int main() 
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		r+=a[i];	//数列之和为右边界 
		l=max(l,a[i]);	//数列最大值为左边界 
	}
	while(l<=r)	//当左边界小于等于右边界 
	{
		mid=(l+r)/2;	//确定中间值 
		if(judge(mid))	//如果最少段数不超过m，寻找左边的mid 
			r=mid-1;
		else	//如果最少段数大于等于m，寻找右边的mid 
			l=mid+1;
	}
	printf("%d\n",l);
	return 0;
}