zzuli 1919 (二分 最大值最小化)
阿新 • • 發佈:2019-02-06
Description
晴天想把一個包含n個整數的序列a分成連續的若干段,且和最大的一段的值最小,但他有強迫症,分的段數不能超過m段,然後他就不會分了。。。他想問你這個分出來的和最大的一段的和最小值是多少?
Input
第一行輸入一個整數t,代表有t組測試資料。
每組資料第一行為兩個整數n,m分別代表序列的長度和最多可分的段數。
接下來一行包含n個整數表示序列。
0<=n<=50000 1<=m<=n,0<=ai<=10000。
Output
輸出一個整數表示和最大的一段的最小值。
Sample Input
1
3 2
1 3 5
Sample Output
5
HINT
1 3 5 分成一段可以為1 3 5和為9,分成兩段可以為1,3 5或者1 3,5,和最大的一段值分別為8,5,所以答案為5
因為n的數值太大,所以只用動態規劃的那個遞推公式是行不通的,所以我們要想其他的方法,這道題利用二分和貪心也就能過了
程式碼
#include<iostream>
#include<stdio.h>
#include<algorithm>
#define LL long long
using namespace std;
const int MAX=50005;
int a[MAX];
int n,m;
bool Judge(int x)
{
int s=0,cnt=0;
for(int i=0; i<n; i++)
{
if(x<a[i])
return false;
if(s+a[i]<=x)
s+=a[i];
else
{
s=a[i];
cnt++;
if(cnt>m-1)
return false;
}
}
return true;
}
LL Solve(LL lo,LL hi)
{
int mid;
while(lo<hi)
{
mid=lo+(hi-lo)/2;
if(Judge(mid))
hi=mid;
else
lo=mid+1;
}
return lo;
}
int main()
{
int T;
LL sum,max;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
max=0;
sum=0;
for(int i=0; i<n; i++)
{
scanf("%d",&a[i]);
if(max<a[i])
max=a[i];
sum+=a[i];
}
printf("%lld\n",Solve(max,sum));
}
return 0;
}