codeforces 1328 F. Make k Equal

题意：

题意
给一个长度为 $n$ 数组，每次可以进行两种操作的任何一个，

1. 选择最大值 -1
2. 选择最小值 +1

如果到达 $k$ 个相同的最少需要多少次操作。

就是对每个位置计算一下，到达它的最小值，求所有位置的最小值。

$$\{\begin{array}{ll}i\ast a_i-{\sum }_1^i{a}_i+(i-k)& i\ge k\\ (a_i\ast i-{\sum }_0^i{a}_i)+\sum N_i-(N-i)\ast a_i-N+K& \\ {\sum }_i^N{a}_i-(N-i+1)\ast a_i-(N-i+1)+K& i+K\le N+1\end{array}$$

此后求一下数组 $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;
}