CodeForces 519D A and B and Interesting Substrings 前缀和

题意：给出26个小写字母的权值，给出一个字符串s，要求找出所有这样的子串，他们的首尾字符相同，并且除了首尾外其余字母首尾之和为0的子串。将数目输出出来。

思路：找前缀和相同，并且字符相同的数目。

简直是黑科技

#include<cstdio>#include<queue>#include<iostream>#include<vector>#include<map>#include<cstring>#include<string>#include<set>#include<stack>#include<algorithm>#define cle(a) memset(a,0,sizeof(a))#define inf(a) memset(a,ox3f,sizeof(a))#define ll long long#define Rep(i,a,n) for(int i=a;i<=n;i++)using namespace std;const int INF = ( 2e9 ) + 2;const int maxn = 28;const int maxm = 1e5+10;map<ll,ll> mp[26];// 找前缀和相同的两个字符 int a[maxn];char s[maxm];int main(){//freopen("in.txt","r",stdin);for(int i=0;i<26;i++)scanf("%d",&a[i]);scanf("%s",s+1);int len=strlen(s+1);ll ans=0,pre=0;for(int i=1;i<=len;i++){int m=s[i]-'a';ans+=mp[m][pre];pre+=a[m];mp[m][pre]++;}printf("%I64d\n",ans);}