CF-Round#629-div3-F题

CF-Round#629-div3-F题

F. Make k Equal

传送门
小白选手的tle之路。
最近div3难度的都可以完成啦~
今天停电从早上我醒来开始停到晚上吃饭。emmm。
手机没电。没网。于是就没写题。。
等到晚上好不容易开始写题之后。。。
肚子疼到起不来。。。
现在肚子好受了些起来补题了。
艰辛qwq

F题是一道纯粹模拟把。。。也叫做贪心题。

题目大意：
给你一个序列。要求序列中要有k个相同的数字。
可以进行两项操作：
把序列中最小的值+1
把序列中最大的值-1

所以我们的贪心策略（噢不对模拟策略）就很明显啦~
这个时候CF有卡住了，我提交了半天没反应。。
记住只有最小的数字才可以加，只有最大的数字才可以减。

我们把数据存到a[]中，然后sort一下。
记录每个数字出现了多少次。
（这里可以用map）
num[]记录每个数
对应的cnt[]记录每个数出现了多少次。

回归正题：
pre[i]代表将小于num[i]的所有数字都变成num[i] - 1的最小步骤。
pcnt[i]代表小于num[i]的个数

suf[i]代表将大于num[i]的所有数字都变成num[i] + 1的最小步骤。
scnt[i]代表大于num[i]的个数

现在我们需要求pre[],pcnt[],suf[],scnt[]；
这很明显。pre[], pcnt[]从前往后（前缀和）
suf[], scnt[]从后往前（后缀和）

重点解释一下两个式子：

考虑pre[i]和pre[i - 1]的关系：
根据上面的定义：pre[i - 1]是表示将所有小于nu[i - 1]的所有数变成num[i - 1] - 1的最小步数。
所以我们在此基础上，只需要求解把处理好的这些数处理成num[i] - 1的步数。
然后再加上把原本没处理的num[i - 1]处理成为num[i] - 1的步数;
这就是第一个式子的含义啦。
pcnt[i] = pcnt[i - 1] + cnt[i - 1]；
这个式子也应该很好理解：
在pcnt[i - 1]的基础上：pcnt[i - 1]代表小于num[i - 1]的个数。这个基础上只要加上num[i - 1]的个数也就是cnt[i - 1]就行。

求后缀和的时候一样的道理：

最后我们只需要跑一遍数组
维护最小值：
res代表还需要多少个数字达到k个要求
res = k - cnt[i];
分情况讨论一下就行：
res <= 0已经达到要求，不需要操作。为0
其余情况就是维护最小的ans就行啦~
要么就是把前面小的数字做+操作，把后面大的数字做-操作。加上需要的个数
要么就是只操作小的数字（前提是小的数字够）
要么就是只操作大的数字（前提是大的数字够）

这应该是不难把~

代码部分：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 10;

int n, k;
int a[N];
int cnt[N];
int num[N];
ll pre[N], pcnt[N];
ll suf[N], scnt[N];

int main()
{
	cin >> n >> k;
	
	for (int i = 1; i <= n; i++)
	{
		cin >> a[i];
	}
	sort(a + 1, a + n + 1);
	int ccnt = 0;
	int m = 0;
	
	for (int i = 1; i <= n; i++)
	{
		ccnt++;
		if (a[i] != a[i + 1])
		{
			num[++m] = a[i];
			cnt[m] = ccnt;
			ccnt = 0;
		}
	}
	for (int i = 2; i <= m; i++)
	{
		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];
	}
	for (int i = m - 1; i >= 1; i--)
	{
		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];
	}
	ll ans = 1e18 + 10;
	for (int i = 1; i <= m; i++)
	{
		int res = k - cnt[i];
		if (res <= 0)
		{
			cout << "0\n";
			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);
		}
	}
	cout << ans << endl;
	return 0;
}