Codeforces Round #294 (Div. 2) D. A and B and Interesting Substrings

题目地址:http://codeforces.com/contest/519/problem/D

    26个字母分别给出喜爱程度，用正负整数表示。然后给出一个只包含英文字母的字符串S，如果T为S子串，T第一个字符与最后一个字符相同，且除第一个和最后一个字符外，字符喜爱程度和为0，则为符合要求的子串。求共有多少符合要求的子串。

    这里如果维护前缀和sum[i]，S[i]=S[k](k>i)，那么中间字符喜爱程度和为0就转换为sum[i]=sum[k]-val[k]。那开个map<LL,int>m[26]就可以了。

#include<cstdio>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<stdbool.h>
#include<map>
#include<stack>
using namespace std;
#define clr(x,y)    memset(x,y,sizeof(x))
#define sqr(x)      ((x)*(x))
#define rep(i,a,b)  for(int i=(a);i<=(b);i++)
#define LL          long long
#define INF         0x3f3f3f3f
#define A           first
#define B           second

int main()
{
    int val[30];
    char a[100005];
    LL rez=0,sum=0;
    
    map<LL,int> m[30];
    for(int i=0;i<26;i++) {
        scanf("%d",&val[i]);
    }
    getchar();
    gets(a);
    
    int len=strlen(a);
    for(int i=0;i<len;i++) {
        a[i]-='a';
        sum+=val[a[i]];
        rez+=m[a[i]][sum-val[a[i]]];
        m[a[i]][sum]++;    
    }
    
    cout<<rez<<endl;
    
}