魔力 - 随机 - 前缀和

题目大意:给定一列数和一个字符集,问有多少子区间每种数字都出现了并且出现的次数都相等。
题解:考虑字符集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);
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值