题意:
给出一个长度为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。因此第一种方案更优。并且第一种方案的最大值是所有方案中最小的。输出这个最小的最大值。
思路:
二分答案
- 就直接二分答案,但是这道题比较。。因为题目并没有说序列是有序的,好烦。。。不过也是一道二分答案的典型例题吧!!!
- 注意:二分的范围,l和r的取值。。。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e4 + 5;
ll n,k;
ll a[maxn];
bool check(ll x){
ll cnt = 1;
ll sum = 0;
for(ll i = 1;i <= n;i++){
if(sum + a[i] > x){
sum = a[i];
cnt++;
}
else{
sum = sum + a[i];
}
}
if(cnt <= k) return true;
else return false;
}
int main(){
scanf("%lld%lld",&n,&k);
memset(a,0,sizeof(a));
for(ll i = 1;i <= n;i++){
scanf("%lld",&a[i]);
}
ll l = 1;
ll r = 5e13;
ll ans;
while(l <= r){
ll mid = (l + r) / 2;
if(check(mid)){
ans = mid;
r = mid - 1;
}
else l = mid + 1;
}
printf("%lld\n",ans);
return 0;
}