KEYENCE Programming Contest 2021 (AtCoder Beginner Contest 227) D - Project Planning（二分+贪心）

Project Planning

题目大意：现给出一个序列 $a_i$ ，代表第 $i$ 种球有多少个。现在认为一个桶内只能放不同种类的球，再给出桶的容量 $k$ ，现在问最多能放满几个桶。

这题的答案显然有二分性，我们可以二分答案解决，本题的关键就在于 $check$ 答案。二分之后我们一定要用到我们二分出来的信息，也就是桶的个数，如果不考虑桶的个数，那么就还是会和没二分一样，不知如何快速分配。

那么我们现在得到了桶的数量 $x$ ，我们可以考虑一下 $a_i\ge x$ 的那些球如何放呢？我们会想，是不是直接每个桶放一个呢？如果就这么放，那么 $a_i<x$ 的那些如何放？这里用代码语言描述更易懂，如下：

set<int> tong[k];
int idx = 0;
for (int i = 1; i < sz; ++i) {
	for (int j = 0; j < a[i]; ++j) {
		tong[idx].insert(a[i][j]);
		idx = (idx + 1) % k;
	}
}

因为 $a_i＜x$ 所以这么放不可能有冲突，也就是说对于 $a_i＜k$ 的元素我们全都可以放下去。那么现在我们现在考虑这是否最优，那么其实我们只要考虑我们这样放是否放到了数量总数的最大值。对于 $a_i\ge k$ 的那些球，最多也就放 $x$ 个，其他的 $a_i<x$ 的还全放下去了，数量已经到极大了，理论上不可能再多了。所以这是最优的情况，那么这里我们只要考虑总数是否大于等于要求的总数 $x\times k$ 即可进行 $check$ 。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 2e5 + 10;

int n;

ll k, a[N];

bool check(ll x) {
	ll sum = 0;
	for (int i = 1; i <= n; ++i) {
		sum += min(x, a[i]);
	}
	return sum >= x * k;
}

int main() {
#ifndef ONLINE_JUDGE
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	scanf("%d%lld", &n, &k);
	for (int i = 1; i <= n; ++i) {
		scanf("%lld", &a[i]);
	}
	ll l = 0, r = 2e17 / k;
	while(l <= r) {
		ll mid = l + r >> 1ll;
		if (check(mid)) l = mid + 1;
		else r = mid - 1;	
	}
	printf("%lld", l - 1);
}