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;
}

转载于:https://www.cnblogs.com/donkey2603089141/p/11416401.html