题目大意:给26个字母赋值,然后输入一个字符串,求首位和末位字符相同且首位和末位之间(不含端点)值的和为0的子串的个数
题目分析:预处理出前缀和,可以发现要满足中间和为0,只需要找前缀和相同的两个字符即可,然后用map离散一下数据,map[字符][对应的前缀和],每次sum加上对应的map值,相当于动态规划的过程
题目分析:预处理出前缀和,可以发现要满足中间和为0,只需要找前缀和相同的两个字符即可,然后用map离散一下数据,map[字符][对应的前缀和],每次sum加上对应的map值,相当于动态规划的过程
#include<stdio.h>
#include<string.h>
#include<map>
#include<algorithm>
using namespace std;
#define LL long long
const LL maxm=1e5+10;
map<LL,LL>q[26];
int a[30];
char s[maxm];
LL p[maxm];
int main()
{
LL sum=0;
memset(p,0,sizeof(p));
for(int i=0;i<26;i++)
{
scanf("%d",&a[i]);
}
scanf("%s",s);
int len=strlen(s);
for(int i=0;i<len;i++)
{
p[i]=p[i-1]+a[s[i]-'a'];
}
for(int i=0;i<len;i++)
{
sum+=q[s[i]-'a'][p[i-1]];
q[s[i]-'a'][p[i]]++;
}
printf("%lld\n",sum);
return 0;
}