2019-2020 ICPC, NERC, Southern and Volga Russian Regional Contest J. The Parade（二分）

题目链接


题意：给定身高为1-n的士兵数量，现在要求士兵分成k排，要求每一排士兵身高差不能超过1，求最多有多少士兵可以拿出来排。
思路：又是一个裸的二分题，但是写check函数的时候确遇到了bug。。。我们可以二分枚举每排士兵的人数。这里就是注意一下，每个身高的士兵人数要累加起来求和，如果分开的话会出错。

#include
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);
	}
}