题目链接
题意:给定身高为1-n的士兵数量,现在要求士兵分成k排,要求每一排士兵身高差不能超过1,求最多有多少士兵可以拿出来排。
思路:又是一个裸的二分题,但是写check函数的时候确遇到了bug。。。我们可以二分枚举每排士兵的人数。这里就是注意一下,每个身高的士兵人数要累加起来求和,如果分开的话会出错。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=3e4+1;
ll n,k,num[maxn],a[maxn];
bool check(ll x)
{
ll cnt=0,sum=0;
for(int i=1;i<=n;++i)
{
if(sum+a[i]<x) sum=0;
cnt+=(sum+a[i])/x;
sum=(sum+a[i])%x;
}
return cnt>=k;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
ll sum=0;
scanf("%lld%lld",&n,&k);
for(int i=1;i<=n;++i) scanf("%lld",&a[i]),sum+=a[i];
if(sum<k)
{
printf("0\n");continue;
}
ll l=1,r=sum/k;
while(l<=r)
{
ll mid=(l+r)>>1;
if(check(mid)) l=mid+1;
else r=mid-1;
}
printf("%lld\n",r*k);
}
}