[题解] [CF 1250J] The Parade

题面

题目大意:

给定一个 (n) , 所有军人的数量均在 ([1, n])

给定 (a_i) 代表高度为 (i) 的军人的个数

你要将这些军人分成 (k) 行, 满足下面两个条件

• 每行人数相等
• 同一行任意两个军人的高度差的绝对值不超过 1

问你最多能够选多少个军人分成 (k) 行

题解

题目满足单调性, 可以二分

我们二分一个 (mid) 表示每一行有 (mid) 个军人

不难证得首先将高度相等的分成一行, 再分高度不相等的是最优的

简要证明 : 可将不是上述方案的方案调整为是上述方案的方案, 结果不会更差

那么现在我们就是要看怎么分是最优的

我们将高度为 (i + 1) 的并到 (i) 上去等价于把 (i) 的并到 (i + 1) 上去

假如说 (a_{i + 1} >= a_i \% mid) , 我们就可以将 (a_{i + 1}) 减去 (a_i \% mod) , 再将 (ans) 加一

正确性:

首先如果不用 (a_i) 剩下的, 和这部分剩下的配对的 (a_{i + 1}) 肯定是要跟自己或者 (a_{i + 2}) 配对的

也就是说这一部分肯定是要用的, 如果用在自己和后面身上, 可能会导致后面的无法配对

也就是说

这一部分总会配对, 跟前面配对的不会对后面造成影响, 因为后面的总数没变

跟后面的配对可能会对后面的答案造成影响, 因为总数变了

由此可见这样配对不会更差

Code

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
typedef long long ll;
const int N = 30005; 
using namespace std;

int T, n; 
ll k, a[N], b[N], ans, sum; 

template < typename T >
inline T read()
{
	T x = 0, w = 1; char c = getchar();
	while(c < '0' || c > '9') { if(c == '-') w = -1; c = getchar(); }
	while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
	return x * w; 
}

bool check(ll x)
{
	ll res = 0; b[1] = a[1]; 
	for(int i = 1; i <= n; i++)
	{
		res += b[i] / x;
		b[i + 1] = a[i + 1]; 
		if(b[i + 1] >= x - (b[i] % x))
			b[i + 1] -= x - (b[i] % x), res++; 
	}
	return res >= k; 
}

int main()
{
	T = read <int> ();
	while(T--)
	{
		ans = sum = 0; n = read <int> (), k = read <ll> ();
		for(int i = 1; i <= n; i++)
			sum += (a[i] = read <ll> ());
		a[n + 1] = 0; 
		ll l = 1, r = sum; 
		while(l <= r)
		{
			ll mid = (l + r) >> 1;
			if(check(mid)) ans = mid, l = mid + 1;
			else r = mid - 1; 
		}
		printf("%lld
", 1ll * ans * k); 
	}
	return 0; 
}