题意:给出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);
}