coderforce 519D A and B and Interesting Substrings

题目大意:给26个字母赋值,然后输入一个字符串,求首位和末位字符相同且首位和末位之间(不含端点)值的和为0的子串的个数
题目分析:预处理出前缀和,可以发现要满足中间和为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;
}

你可能感兴趣的:(coderforce 519D A and B and Interesting Substrings)