Codeforces Round #294 (Div. 2) D. A and B and Interesting Substrings(前缀和＋map)

sum数组表示区间前缀和。

如果位置i+1到位置j-1的和为零，那么sum[i]=sum[j-1],既sum[i]+a[k]=sum[j] （i和j位置字母为k）

这样的话对每个字母枚举所有位置做起终点是可以的，但是时间复杂度为N^2。

考虑用一个map维护前缀和中每个值出现的次数，这样可以用log时间查看某个值的前缀和存不存在。

代码：

//
//  main.cpp
//  294D. A and B and Interesting Substrings
//
//  Created by Baoli1100 on 15/2/28.
//  Copyright (c) 2015年 Baoli1100. All rights reserved.
//

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#include <vector>
#include <map>
#define LL long long
LL sum[100005];
int a[30];
vector<int> pos[30];
char s[100005];
map<LL ,int > Cur;
int main(){
    for(int i=0;i<26;i++){
        scanf("%d",&a[i]);
    }
    scanf("%s",s);
    int Len=strlen(s);
    LL res=0;
    for(int i=0;i<Len;i++){
        int t=s[i]-'a';
        pos[t].push_back(i);
        if(i==0){
            sum[i]=a[t];
        }
        else sum[i]=sum[i-1]+a[t];
    }
    for(int i=0;i<26;i++){
        Cur.clear();
        for(int j=0;j<pos[i].size();j++){
            int tmp=pos[i][j];
            Cur[sum[tmp]]++;
        }
        for(int j=0;j<pos[i].size();j++){
            int tmp=pos[i][j];
            Cur[sum[tmp]]--;
            res+=Cur[sum[tmp]+a[i]];
        }
    }
    printf("%lld",res);
    return 0;
}