JXOI 2017 颜色 题解

T3 颜色 100/100

对于这题由于数据范围小，有一种神奇的做法，我们可以把每个值随机赋值，但是保证相同颜色的和为0，就代表消去了这个颜色，我们只要取寻找合法区间就行，合法区间的寻找只要找相同前缀和就行，因为相同前缀和之间这一段为0，也就代表这一段是合法区间。

代码

#include<bits/stdc++.h>
#include<ctime>
using namespace std;
const long long mod=1e9+7;
const int N=3e5+10;
long long num[N];
int T,n,a[N],f[2]={1,-1},k;
map<long long,long long> book;
vector <int> p[N];
long long x,add,flag,ans;
int main(){
	freopen("color.in","r",stdin);
	freopen("color.out","w",stdout);
	srand((unsigned)time(0));
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		for(int i=1;i<=n;++i){
			scanf("%d",&a[i]);
			p[a[i]].push_back(i);
		}
		for(int i=1;i<=n;++i){
			if(p[i].size()==0) continue;
			else if(p[i].size()==1) num[p[i][0]]=0;
			else{
				add=0;
				for(int j=0;j<p[i].size()-1;++j){
					k=rand()&1;
					x=(rand()*rand()%mod*rand()%mod*rand()%mod)%mod*f[k];
					num[p[i][j]]=x;
					add+=x;
				}
				num[p[i][p[i].size()-1]]=-add;
			}
		}
		book.clear();add=0;book[0]=1;ans=0;
		for(int i=1;i<=n;++i){
			add+=num[i];
			ans+=book[add];
			++book[add];
		}
		printf("%lld\n",ans);
		for(int i=1;i<=n;++i){
			p[i].clear();
		}
	}
	return 0;
}