题目大意:给定一列数和一个字符集,问有多少子区间每种数字都出现了并且出现的次数都相等。
题解:考虑字符集2怎么做,分别看为1和-1然后区间和等于0即可。考虑给每个数值赋一个随机权值,并所有种类的数的随即权值和为0,然后统计区间和为0的区间数即可。使用unordered_map可以线性。
#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define Rep(i,v) rep(i,0,(int)v.size()-1)
#define lint long long
#define mod 1000000007
#define ull unsigned lint
#define db long double
#define pb push_back
#define mp make_pair
#define fir first
#define sec second
#define gc getchar()
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
typedef pair<int,int> pii;
typedef set<int>::iterator sit;
inline int inn()
{
int x,ch;while((ch=gc)<'0'||ch>'9');
x=ch^'0';while((ch=gc)>='0'&&ch<='9')
x=(x<<1)+(x<<3)+(ch^'0');return x;
}
const int N=100010,SIG=100;
char str[N];int vis[SIG],lst[SIG];
ull val[SIG],s[N];unordered_map<ull,int> psc;
inline int toi(char c) { return c>='a'&&c<='z'?c-'a'+1:c-'A'+1+26; }
struct Rand{
ull x;Rand() { x=0; }
inline ull operator()() { return ((x*=998244353)+=1000000007)^=19260817; }
}rnd;
int main()
{
int n=inn(),c,cnt=0;lint ans=0;
scanf("%s",str+1);
rep(i,1,n) if(!vis[c=toi(str[i])]) vis[c]=1,val[c]=rnd(),lst[++cnt]=c;
c=lst[cnt],val[c]=0;rep(i,1,cnt-1) val[c]-=val[lst[i]];
rep(i,1,n) s[i]=s[i-1]+val[toi(str[i])];
psc[0]=1;
rep(i,1,n)
{
if(psc.count(s[i])) ans+=psc[s[i]];
psc[s[i]]++;
}
return !printf("%lld\n",ans%mod);
}