BZOJ 3969 low power

Description

Input

Output

Sample Input

Sample Output

HINT

\(2nk \le 10^{6}, 1 \le p_{i} \le 10^{9}\)。

答案明显满足可二分性，我们可以利用贪心检验。

我们将所有的\(power\)从小到大排个序。然后二分一个结果\(mid\)，从左往右枚举\(i\)，如果\(power_{i+1}-power_{i} \le mid\)，就选择这两个电池作为能量最小的电池装到一个机器人上，直到选出\(2n\)个。最后我们只需要检验是否每个电池都可以在它后面跟上\(K-1\)个电池即可。这个从左往右for一遍贪心即可。

#include<algorithm>
#include<cstdio>
#include<cstdlib>
using namespace std;

#define maxn (1000010)
int N,K,power[maxn],cho[maxn];

inline bool check(int key)
{
int nn = 0;
for (int i = 1;i < (N*K<<1)&&nn < (N<<1);++i) if (power[i+1]-power[i] <= key) cho[++nn] = i,cho[++nn] = i+1,++i;
if (nn != (N<<1)) return false;
for (int i = nn,last = (N*K<<1)+1,tot = 0;i;--i)
{
tot -= cho[i]-last; tot -= K; last = cho[i];
if (tot < 0) return false;
}
return true;
}

int main()
{
freopen("3969.in","r",stdin);
freopen("3969.out","w",stdout);
scanf("%d %d",&N,&K);
for (int i = 1;i <= (N*K<<1);++i) scanf("%d",power+i);
sort(power+1,power+(N*K<<1)+1);
int l = 0,r = 1<<30;
while (l <= r)
{
int mid = (l + r) >> 1;
if (check(mid)) r = mid - 1;
else l = mid + 1;
}
printf("%d",l);
fclose(stdin); fclose(stdout);
return 0;
}