[ZJOI2022]众数

最新推荐文章于 2023-07-17 12:34:21 发布

StaroForgin

最新推荐文章于 2023-07-17 12:34:21 发布

阅读量583

点赞数 1

分类专栏： ------动态规划------ 文章标签：算法

本文链接：https://blog.csdn.net/Tan_tan_tann/article/details/124666293

版权

------动态规划------ 专栏收录该内容

174 篇文章 0 订阅

订阅专栏

众数

题解

~~听 $\rm F\color{red}{utaRimeWoawaSete}$ 说了说这道题，就去做了一下。说不定是今年ZJOI唯一一道签到题。~~

由于它要输出可能的最大值并将所有可能的最大值都输出，那么我们显然都要尽可能求出某个数可能的最多出现次数。
由于我们的 $a_i$ 并不全相等，所以我们最后的可能成为众数的值一定是一个原先出现过的数，而且这个值必然是将原来一个区间 $(l, r)$ 中的数 $x$ 变到区间外的数 $y$ 而得到的。
如果只是某一个数变成的话，我们不如不变那一个数，将其它数变到它身上，这样反而能够得到更大的答案。

如果数 $x$ 出现的位置时 ${p_0=0,p_1,p_2,...,p_{k-1},p_k=n+1\}$ ，记区间 $(l, r)$ 中出现次数最多的数出现了 $c (l, r)$ 次，那么 $x$ 最后的答案便是 $\max_{0\leqslant i\leqslant j\leqslant k}(k-j+i+c(p_i,p_j))$ 。
一种可行的方法是枚举出现在 $p_i,p_j)$ 中出现次数最多的数为 $x$ 。
记前缀 $i$ 中出现了 $pre_i$ 个 $x$ ，显然，我们可以改写一下上面的式子，
$ans_x=k+\max_{0\leqslant i\leqslant j\leqslant k} \left((pre_{p_i}+i)-(pre_{p_j}+j)\right)$ 这样不就只需要记录下一个前缀最小的 $pre_{p_i}+i$ ，就行了吗？
于是我们就得到了一个 $O\left(n^2\right)$ 做法了吗，但显然是过不了 $n\leqslant 2\times 10^5$ 的数据的。

没事，我们继续想想上面方法的瓶颈主要是在什么地方。
显然这个方法我们对于每个颜色作为中间的数都要将整个序列扫一遍，而许多出现次数很少的颜色其实是完全没必要扫一遍的。
不如考虑根号分治，我们对出现次数很少的颜色想另外的做法。
由于这些颜色出现不超过 $B$ 次，不妨就枚举我们区间内出现最多的颜色出现 $i\in[1,B]$ 次时的答案。
显然，对于每个左端点，当右端点为 $r$ 时，最右边的左端点 $l$ 使得 $(l, r)$ 中有颜色出现 $i$ 次的这个 $l$ 显然是关于 $r$ 单增的。
这样我们就可以只需要知道在 $[1,l]\cup[r,n]$ 中出现次数最多的颜色的出现次数就行。
显然，对于任意的一种颜色，我们让 $a_r$ 刚好为这种颜色显然是最优的，那么我们需要知道的其实是 $[1,l]\cap[r,n]$ 中出现了多少个颜色 $a_r$ 即可。
这个由于我们的 $r$ 是不断右移的，所以我们也可以对颜色 $a_r$ 设计一个指针，可以用前向星维护，让他跑到在 $l$ 右边的最后一个 $a_r$ 处，这样就可以很简单的统计这东西了。
我们也就用 $O\left(nB\right)$ 的时间完成了出现次数少于 $B$ 的颜色的答案统计。

总时间复杂度 $O\left(\frac{n^2}{B}+nB\geqslant n\sqrt{n}\right)$ ，让 $B$ 取 $\sqrt{n}$ 时最优。

源码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int> pii;
#define MAXN 200005
#define MAXM (1<<18)+5
#define pb push_back
#define mkpr make_pair
#define fir first
#define sec second
const int INF=0x3f3f3f3f;
const int mo=1e9+7;
const int iv2=5e8+4;
const int lim=1000000;
const int n1=300;
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
int add(int x,int y,int p){return x+y<p?x+y:x+y-p;}
void Add(int &x,int y,int p){x=add(x,y,p);}
int qkpow(int a,int s,int p){int t=1;while(s){if(s&1)t=1ll*a*t%p;a=1ll*a*a%p;s>>=1;}return t;}
int T,n,a[MAXN],b[MAXN],cnt[MAXN],maxx[MAXN];
int pre[MAXN],nxt[MAXN],head[MAXN],dp[MAXN];
int tot,buc[MAXN],pos[MAXN],id[MAXN],ans;
int main(){
	read(T);
	while(T--){
		read(n);for(int i=1;i<=n;i++)read(a[i]),b[++tot]=a[i];
		sort(b+1,b+tot+1);tot=unique(b+1,b+tot+1)-b-1;
		for(int i=1;i<=n;i++)a[i]=lower_bound(b+1,b+tot+1,a[i])-b;
		for(int i=1;i<=n;i++)cnt[a[i]]++,id[i]=cnt[a[i]];
		for(int i=1;i<=tot;i++)maxx[i]=cnt[i];
		for(int i=n;i>0;i--)nxt[i]=head[a[i]],pre[head[a[i]]]=i,head[a[i]]=i;
		for(int i=1;i<=tot;i++)if(cnt[i]>n1)
			for(int j=1,num=0;j<=n;j++){
				const int x=a[j];
				if(x==i)num++;
				else maxx[x]=max(dp[pre[j]]+num+cnt[x]-id[j]+1,maxx[x]),
					dp[j]=max(dp[pre[j]],id[j]-num),
					maxx[x]=max(maxx[x],dp[j]+cnt[i]);
			}
		for(int i=min(n1,n);i>0;i--){
			int num=0,l=1;
			for(int j=1;j<=tot;j++)buc[j]=0,pos[j]=head[j];
			for(int j=1;j<=n;j++){
				const int x=a[j];buc[x]++;if(buc[x]==i)num++;
				while(l<=j&&num>(buc[a[l]]==i)){
					if(buc[a[l]]==i)num--;
					buc[a[l]]--;l++;
				}
				while(nxt[pos[x]]&&nxt[pos[x]]<l)pos[x]=nxt[pos[x]];
				int tmp=cnt[x]-id[j]+1+id[pos[x]]-(pos[x]>=l);
				if(num)maxx[x]=max(maxx[x],tmp+i);
			}
			for(int j=1;j<=tot;j++){
				while(nxt[pos[j]]&&nxt[pos[j]]<l)pos[j]=nxt[pos[j]];
				if(num)maxx[j]=max(maxx[j],id[pos[j]]+i-(pos[j]>=l));
			}
		}
		int mx=0;
		for(int i=1;i<=tot;i++)mx=max(mx,maxx[i]);
		printf("%d\n",mx);
		for(int i=1;i<=tot;i++)if(maxx[i]==mx)printf("%d\n",b[i]);
		for(int i=1;i<=tot;i++)head[i]=cnt[i]=0;tot=0;
		for(int i=1;i<=n;i++)pre[i]=nxt[i]=id[i]=0;
	}
	return 0;
}