codeforces1447F Frequency Problem

题目：

给定一个长度为$n$的序列$a$，求最长的连续子序列，满足出现次数最多的数的种类至少为两种，输出其长度。
$(1\le n\le 200000)$

• easy version

  题目链接
  $(1\le a_i\le min(n,100))$

  • 题解：

    结论：在原序列$a$中出现次数最多的数$p$在题目要求的连续子列中也一定出现次数最多。
    证明：
    （1）如果序列$a$中出现次数最多的数的种类不止一种，那么序列$a$就是要求的最长序列，上述结论成立。
    （2）如果序列$a$中出现次数最多的数的种类只有一种，设为$p$，那么我们从两边开始收缩，在这个过程中，如果存在$v$，它的出现次数和$p$相同，那么当前区间的序列有可能称为满足题意的序列，这种情况下就不用再收缩了，因为再收缩一定不是最长的了，否则不存在这样的$v$，那么$p$仍然是唯一的出现次数最多的数，还需要再收缩。所以在整个收缩过程中出现的所有可能满足题意的序列，上述结论成立。

    基于上述结论，我们可以$O(n)$找出$p$，由于数的种类很少，枚举所有组合$(p,v)$，然后$O(n)$找到满足$nu{m}_p=nu{m}_v$的最长区间。具体怎么找一对$(p,v)$的最长区间，构造一个新的序列$b$，如果$a_i=p$，则$b_i=1$，如果$a_i=v$，则$b_i=-1$，其他位置$b_i=0$，得到$b$前缀和序列$sum$，如果$su{m}_i=su{m}_j$，那么$(i,j]$是一个满足$nu{m}_v=nu{m}_p$的区间，找到每个$sum$值对应的最小位置和最大位置，更新答案即可。

  • 复杂度：$O(100n)$

  • 代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<string>
#include<bitset>
#include<sstream>
#include<ctime>
//#include<chrono>
//#include<random>
//#include<unordered_map>
using namespace std;

#define ll long long
#define ls o<<1
#define rs o<<1|1
#define pii pair<int,int>
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define sz(x) (int)(x).size()
#define all(x) (x).begin(),(x).end()
const double pi=acos(-1.0);
const double eps=1e-6;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const int maxn=200005;
ll read(){
	ll x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int n;
int a[maxn],cnt[105],sum[maxn];
pii buf[2*maxn];
int main(void){
	// freopen("in.txt","r",stdin);
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		cnt[a[i]]++;
	}
	int p,mx=0,num=0;
	for(int i=1;i<=100;i++){
		if(cnt[i]>mx){
			mx=cnt[i];
			p=i;
			num=1;
		}
		else if(cnt[i]==mx){
			num++;
		}
	}
	if(num!=1){
		printf("%d\n",n);
	}
	else if(mx==n){
		puts("0");
	}
	else{
		int ans=0;
		for(int i=1;i<=100;i++){
			if(i!=p&&cnt[i]){
				//[-n,n]:0->2n
				for(int j=0;j<=2*n+1;j++){
					buf[j].fi=INF;
					buf[j].se=0;
				}
				buf[n].fi=0;
				for(int j=1;j<=n;j++){
					sum[j]=sum[j-1];
					if(a[j]==p){
						sum[j]++;
					}
					else if(a[j]==i){
						sum[j]--;
					}
					buf[sum[j]+n].fi=min(buf[sum[j]+n].fi,j);
					buf[sum[j]+n].se=j;
				}
				for(int j=0;j<=2*n;j++){
					ans=max(ans,buf[j].se-buf[j].fi);
				}
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}

• hard version

  题目链接
  $(1\le a_i\le n)$
  暂时不会，咕