51nod-1128 正整数分组 V2

1128 正整数分组 V2 

基准时间限制：1 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

 收藏

 关注

给出一个长度为N的正整数数组，不改变数组元素的顺序，将这N个数分为K组。各组中元素的和分别为S1,S2....Sk。如何分组，使得S1至Sk中的最大值最小？

例如：1 2 3 4 5 6分为3组，{1 2 3} {4 5} {6}，元素和为6, 9, 6，最大值为9。也可以分为{1 2 3 4} {5} {6}。元素和为：10 5 6，最大值为10。因此第一种方案更优。并且第一种方案的最大值是所有方案中最小的。输出这个最小的最大值。

Input

第1行：2个数N, K，中间用空格分隔，N为数组的长度，K为要分为多少组。(2 <= K < N <= 50000)
第2 - N + 1行：数组元素(1 <= A[i] <= 10^9)

Output

输出这个最小的最大值。

Input示例

6 3
1
2
3
4
5
6

Output示例

9

思路：这题可以用dp来做，当时间复杂度 O(n*k)会超时，此时可以考虑 nlog(n)的复杂度，对于 区间 a[1->n] 要分成 k个区间，求所有情况的 k个区间和的最大值的最小值 ans，那么如何去分成k区间，按k区间的分界点去考虑的话，肯定会超时，因此应该换个方向考虑。由于题目要求 ans,那么可以 枚举ans来求解, ans范围从 1到a[1->n]总和ans。 假设 ans为 p,那么可以 看 a[]最小可以分成 多少区间，如果 区间数t <k，则说明 p>ans ，如果 t>k ,则 p<ans， 若 t=k，那么说明 p>=ans. 而 对于 ans的枚举可以 用 二分搜索 优化。因此时间复杂度可 优化为 nlong(sum)= 10^6

Code:

#include<iostream>
using namespace std;
typedef long long LL;

const int MAX_N=50005;
int n,m;
int a[MAX_N];
LL ans,sum;

int f(LL p);
int main()
{
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;++i)
	{
		cin>>a[i];
		sum+=a[i];
	}
	LL l=1,h,r=sum;
	while(l<=r){
		h=(l+r)/2;
		if(f(h)==1)	r=h-1;
		else	l=h+1;
	}
	ans=l;
	cout<<ans<<endl;
	return 0;
}

int f(LL p)
{
	int k=1;
	LL s=0;
	for(int i=1;i<=n;++i)
		if(s+a[i]>p){
			s=a[i];	k++;
		}else	s+=a[i];
	if(k<=m)	return 1;
	else	return -1;
}