「EZEC-14」众数 I
题目背景
pigstd 是一个可爱的男孩子。他在 NOI2022 中的众数一题定义了
1
0
6
10^6
106 个 std::deque
并没有 MLE。
题目描述
给定一个长度为 n n n 的序列 a a a,我们通过以下方式构造序列 b b b:
- 初始时 b = a b=a b=a。
- 依次对 b b b 进行 k k k 次操作,每次操作选择任意一个元素并将其修改为任意整数。
dXqwq 定义一个序列的众数为所有出现次数最大的数。例如 [ 1 , 1 , 4 , 5 , 1 , 4 ] [1,1,4,5,1,4] [1,1,4,5,1,4] 的众数为 1 1 1,而 [ 1 , 14 , 5 , 14 , 19 , 19 , 8 , 10 ] [1,14,5,14,19,19,8,10] [1,14,5,14,19,19,8,10] 的众数为 14 , 19 14,19 14,19。
你需要求出有多少整数可能成为 b b b 的众数。
输入格式
第一行输入两个整数 n , k n,k n,k。
第二行输入 n n n 个整数 a i a_i ai。
输出格式
输出一个整数,代表可能成为众数的数的数量。
特别地,如果答案为正无穷,输出 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 1,2,3,4,5 1,2,3,4,5 可能为区间众数。
对于第二组数据,将第一个数换成 6 , 7 , 8 , 9 , ⋯ 6,7,8,9,\cdots 6,7,8,9,⋯ 后它们均会成为区间众数,因此答案为正无穷。
对于第三组数据, 1 , 2 , 3 1,2,3 1,2,3 可能成为区间众数。
【提示】
开
1
0
6
10^6
106 个 std::deque
在空间限制为 1024MB 时不一定会 MLE。
【数据范围】
本题采用捆绑测试。
- Subtask 1(20 pts): n ≤ 5 n\leq 5 n≤5。
- Subtask 2(20 pts): n ≤ 1 0 3 n\leq 10^3 n≤103。
- Subtask 3(20 pts): k = 0 k=0 k=0。
- Subtask 4(20 pts): k = 1 k=1 k=1。
- Subtask 5(20 pts):无特殊限制。
对于
100
%
100\%
100% 的数据,
1
≤
n
≤
1
0
6
1\leq n\leq 10^6
1≤n≤106,$0\leq k\leq n $,
1
≤
a
i
≤
n
1\leq a_i\leq n
1≤ai≤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;
}