P9460 「EZEC-14」众数 I [贪心详细过程]

「EZEC-14」众数 I

题目背景

pigstd 是一个可爱的男孩子。他在 NOI2022 中的众数一题定义了 $10^6$ 个 std::deque 并没有 MLE。

题目描述

给定一个长度为 $n$ 的序列 $a$，我们通过以下方式构造序列 $b$：

• 初始时 $b=a$。
• 依次对 $b$ 进行 $k$ 次操作，每次操作选择任意一个元素并将其修改为任意整数。

dXqwq 定义一个序列的众数为所有出现次数最大的数。例如 $[1,1,4,5,1,4]$ 的众数为 $1$，而 $[1,14,5,14,19,19,8,10]$ 的众数为 $14,19$。

你需要求出有多少整数可能成为 $b$ 的众数。

输入格式

第一行输入两个整数 $n,k$。

第二行输入 $n$ 个整数 $a_i$。

输出格式

输出一个整数，代表可能成为众数的数的数量。

特别地，如果答案为正无穷，输出 pigstd。

样例 #1

样例输入 #1

5 0
1 2 3 4 5

样例输出 #1

5

样例 #2

样例输入 #2

5 1
1 2 3 4 5

样例输出 #2

pigstd

样例 #3

样例输入 #3

5 1
1 1 2 2 3

样例输出 #3

3

提示

【样例解释】

对于第一组数据，最终 $1,2,3,4,5$ 可能为区间众数。

对于第二组数据，将第一个数换成 $6,7,8,9,\cdots$ 后它们均会成为区间众数，因此答案为正无穷。

对于第三组数据，$1,2,3$ 可能成为区间众数。

【提示】

开 $10^6$ 个 std::deque 在空间限制为 1024MB 时不一定会 MLE。

【数据范围】

本题采用捆绑测试。

• Subtask 1（20 pts）：$n\le 5$。
• Subtask 2（20 pts）：$n\le 10^3$。
• Subtask 3（20 pts）：$k=0$。
• Subtask 4（20 pts）：$k=1$。
• Subtask 5（20 pts）：无特殊限制。

对于 $100\%$ 的数据，$1\le n\le 10^6$，$0\leq k\leq n $，$1\le a_i\le n$。
贪心思路:
先开一个ans数值来存放每一个数出现的次数,因为是看众数所以数本身不重要,重要的是出现的次数,然后将这个数组放入优先队列–大根堆里面,它的作用是每次取出最大值然后,为什么用大根堆呢,因为我们需要每次操作都取得最大值,我们这里是该数,如果俩不同的数都出现4次,其中一个-1变成3,那此时的最大数就是另一个数了,所以这一点要清楚:
大根堆:priority_queue<int, vector<int>, less<int>>q;
代码实现:

#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#include<queue>
using namespace std;
int n, m, mmax, res, k=1, numm, ssum;
int w[1000005];
int num[1000005];
int ans[1100005];
bool st[1100005];
int s[1000005];
int main()
{
	priority_queue<int, vector<int>, less<int>>q;
	cin >> n >> m;
	int tm = m;
	for (int i = 1; i <= n; i++) {
		cin >> w[i];
		num[w[i]]++;//记录次数
	}
	for (int i = 1; i <= n; i++) {
		if (!st[w[i]]) {//没被存过
			ans[k] = num[w[i]];//把次数数组存入
			k++;
			st[w[i]] = 1;//表示这个数存过了
		}
	}
	k--;
	sort(ans + 1, ans + 1 + k);//排个序
	mmax = ans[k];//最大
	for (int i = 1; i <= k; i++) q.push(ans[i]);//放入优先队列
	while (tm--)
	{
		int x = q.top() - 1;//最大减一
		q.pop();//弹出去
		q.push(x);//再回来
	}
	int temp = q.top(), res = 0;//记录最大次数
	if (temp <= m)
	//当前的最大次数抵不过m,意味着m次修改我们可以为所欲为
	{
		cout << "pigstd";
		return 0;
	}
	for (int i = 1; i <= k; i++) {
		if (ans[i] + m >= temp) {//可以改
			res++;//答案+1
		}
	}
	cout << res << endl;
	
	return 0;
}