[THUPC 2023 初赛] 众数

P9143 [THUPC 2023 初赛] 众数

题目描述

你有若干个 $[1,n]$ 内的正整数：对于 $1\le i\le n$，你有 $a_i$ 个整数 $i$。设 $S={\sum }_{i=1}^n{a}_i$。

对于一个序列 $p_1,p_2,\cdots ,p_l$，定义其众数 $\text{maj}(p_1,p_2,\cdots ,p_l)$ 为出现次数最多的数。若有多个数出现次数最多，则其中最大的数为其众数。

现在你需要把这 $S$ 个数排成一个序列 $b_1,b_2,\cdots ,b_S$，使得 ${\sum }_{i=1}^S\text{maj}(b_1,b_2,\cdots ,b_i)$ 最大。输出该最大值。

题解

设当前满足$a_i\ne 0$的最大的$i$值为$k$，则对于$1\le i\le k$，我们可以让尽量多的$i$值放入序列，且$i$的数量不超过$a_k$。那么此时答案中$a_k$出现的次数为$\sum _{i=1}^k\min (a_i,a_k)$。每个位置也要减去对应的值，即$a_i=a_i-\min (a_i,a_k)$。

那么，对于所有的$i$，$a_i$可以分成若干段，每一段对应不同$a$值的贡献。我们可以$O(n)$求出每一段的长度并用队列维护。

用桶来存每个$a_i$可取到的值，根据队列的每一段来统计其贡献。

可以根据代码来帮助理解。时间复杂度为$O(n+mxa)$，其中$mxa$为$a_i$的最大值。

code

#include<bits/stdc++.h>
using namespace std;
int n,mx,vk=1,a[100005],v[100005],z[100005];
long long ans=0;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);++z[a[i]];
		mx=max(mx,a[i]);
	}
	v[++v[0]]=n;
	for(int i=n-1;i>=1;i--){
		if(a[i]>a[v[v[0]]]){
			v[++v[0]]=i;
		}
	}
	for(int i=mx;i>=1;i--) z[i]+=z[i+1];
	for(int i=1;i<=mx;i++){
		ans+=1ll*z[i]*v[vk];
		if(i==a[v[vk]]) ++vk;
	}
	printf("%lld",ans);
	return 0;
}