题意:
题意
给一个长度为
n
n
n 数组,每次可以进行两种操作的任何一个,
- 选择最大值 -1
- 选择最小值 +1
如果到达 k k k 个相同的最少需要多少次操作。
就是对每个位置计算一下,到达它的最小值,求所有位置的最小值。
{ i ∗ a i − ∑ 1 i a i + ( i − k ) i ≥ k ( a i ∗ i − ∑ 0 i a i ) + ∑ N i − ( N − i ) ∗ a i − N + K ∑ i N a i − ( N − i + 1 ) ∗ a i − ( N − i + 1 ) + K i + K ≤ N + 1 \begin{cases} i∗a_i−\sum_1^ia_i+(i−k) & i≥k \\ (a_i∗i−\sum_0^ia_i)+\sum N_i−(N−i)∗a_i−N+K\\ \sum_i^N a_i−(N−i+1)∗a_i−(N−i+1)+K & i+K≤N+1 \\ \end{cases} ⎩⎪⎨⎪⎧i∗ai−∑1iai+(i−k)(ai∗i−∑0iai)+∑Ni−(N−i)∗ai−N+K∑iNai−(N−i+1)∗ai−(N−i+1)+Ki≥ki+K≤N+1
此后求一下数组 a a a 的前缀和,对每个进行优化,取最小值。
AC代码:
const int N = 2e5 + 10;
ll a[N], num[N], cnt[N];
ll pre[N], suf[N], pcnt[N], scnt[N];
ll ans = 1e18;
ll n, m, k;
ll res, tmp;
int main()
{
m = 0;
sldd(n, k);
rep(i, 1, n)
sld(a[i]);
sort(a + 1, a + 1 + n);
res = 0;
rep(i, 1, n)
{
res++;
if (a[i] != a[i + 1])
{
m++;
num[m] = a[i];
cnt[m] = res;
res = 0;
}
}
rep(i, 2, m)
{
pre[i] = pre[i - 1] + pcnt[i - 1] * (num[i] - num[i - 1]) + cnt[i - 1] * (num[i] - num[i - 1] - 1);
pcnt[i] = pcnt[i - 1] + cnt[i - 1];
}
per(i, m - 1, 1)
{
suf[i] = suf[i + 1] + scnt[i + 1] * (num[i + 1] - num[i]) + cnt[i + 1] * (num[i + 1] - num[i] - 1);
scnt[i] = scnt[i + 1] + cnt[i + 1];
}
rep(i, 1, m)
{
res = k - cnt[i];
if (res <= 0)
{
puts("0");
return 0;
}
if (i > 1 && i < m)
ans = min(ans, pre[i] + suf[i] + res);
if (i > 1 && pcnt[i] >= res)
ans = min(ans, pre[i] + res);
if (i < m && scnt[i] >= res)
ans = min(ans, suf[i] + res);
}
pld(ans);
return 0;
}