Codeforces 519D A and B and Interesting Substrings(二维map+前缀和)

题目链接：http://codeforces.com/problemset/problem/519/D

题目大意：
给你一串字符串s仅由小写字母组成,并且对于'a'~'z'都给了一个值。求子串t满足t的开头和结尾字符相同，且中间的字符的值相加为0，
求子串t的数目。
解题思路：
设置map<LL,int>mp[26]这样的二维map，记录对应以每个字母结尾的前缀和的值的出现次数。
然后计算前缀和sum，每次计算sum+val[s[i]]前，先计算ans，因为sum[i-1]-sum[x]==0才能说明sum(i-1~x+1)这一段为0 。

代码

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #include<vector>
 #include<string>
 #include<string.h>
 #include<cctype>
 #include<math.h>
 #include<stdlib.h>
 #include<stack>
 #include<queue>
 #include<set>
 #include<map>
 #define lc(a) (a<<1)
 #define rc(a) (a<<1|1)
 #define MID(a,b) ((a+b)>>1)
 #define fin(name)  freopen(name,"r",stdin)
 #define fout(name) freopen(name,"w",stdout)
 #define clr(arr,val) memset(arr,val,sizeof(arr))
 #define _for(i,start,end) for(int i=start;i<=end;i++)
 #define FAST_IO ios::sync_with_stdio(false);cin.tie(0);
 using namespace std;
 typedef long long LL;
 const int N=1e5+;
 const LL INF64=1e18;
 const int INF=0x3f3f3f3f;
 const double eps=1e-;

 int val[];
 map<LL,int>mp[];

 int main(){
     for(int i=;i<;i++){
         cin>>val[i];
     }
     string str;
     cin>>str;
     LL sum=,ans=;
     for(int i=;i<str.length();i++){
         int ch=str[i]-'a';
         //注意先计算ans, 再sum+=val[ch]
         //因为sum[i-1]-sum[x]==0才能说明sum(i-1~x+1)这一段为0
         ans+=mp[ch][sum];
         sum+=val[ch];
         mp[ch][sum]++;
     }
     cout<<ans<<endl;
     return ;
 }