P4065 [JXOI2017]颜色

题面

　　• get到一种 hash 乱搞。

　　• 考虑题目要求如何在 hash 值中得以体现。

　　• 为每个数赋上一个 hash 值，使得同一类数的 hash 值为零。

　　• 那么满足要求的序列的 hash 值之和一定为零。

#include<bits/stdc++.h>
#define ll unsigned long long
using namespace std;
int t,n,x,tot;
int    head[300050];
int nex[300050];
int ver[300050];
ll s[300050];
ll ans,tmp;
void add(int x,int y)
{
    nex[++tot]=head[x];
    ver[tot]=y;
    head[x]=tot;
}
ll randm()
{
    ll cnt=rand()*rand();
    if(rand()&1)    return cnt;
    else    return -cnt;
}
int main()
{
    srand(time(0));
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);tot=0;
        memset(head,0,sizeof(head));
        for(int i=1;i<=n;++i)
        {
            scanf("%d",&x);
            add(x,i);
        }
        for(int i=1;i<=n;++i)
        {
            tmp=0;
            for(int j=nex[head[i]];j;j=nex[j])
            {
                s[ver[j]]=randm();
                tmp-=s[ver[j]];
            }
            s[ver[head[i]]]=tmp;
        }
        s[0]=0;ans=0;
        for(int i=1;i<=n;++i)
            s[i]+=s[i-1];
        sort(s,s+n+1);
        for(int i=0,j=0;i<=n;++j,i=j)
        {
            while(s[j+1]==s[i]&&j<n)
                ++j;
            ans+=(ll)(j-i)*(j-i+1)/2;
        }
        printf("%lld
",ans);
    }
    return 0;
}