比赛 2024暑假C班集训5 评测结果 AATTTTTTTT
题目名称 充电宝 最终得分 20
用户昵称 蜀山鸭梨大 运行时间 8.019 s
代码语言 C++ 内存使用 8.17 MiB
提交时间 2024-07-05 09:51:22
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
int n,ans;
int a[200005],b[200005],l[200005],r[200005];
vector <int> v;
int main(){
	freopen("charger.in","r",stdin);
	freopen("charger.out","w",stdout);
	scanf("%d",&n);
	v.push_back(0);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		if(!b[a[i]]){
			b[a[i]]++;
			v.push_back(a[i]);
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<v.size();j++){
			b[v[j]]++;
		}
		r[i-b[a[i]]]=b[a[i]];
		l[i]=b[a[i]];
		b[a[i]]=0;
	}
	for(int i=1;i<=n;i++){
		int R;
		if(r[i]) R=r[i];
		else R=200005;
		for(int j=1;j<=R&&i+j<=n;j++){
			if(l[i+j]>j){
				ans++;
			}
		}
	}
	printf("%d",ans);
	return 0;
}