#294 (div.2) D.A and B and Interesting Substring

1.题目描述：点击打开链接

2.解题思路：本题要求在s中找满足下列条件的字串：(i)i+1~j-1的字符串权值之和为0；(ii)s[i]==s[j]；处理带权字符串的一般解法是利用前缀和。但我最初的算法效率十分低下：先求出前i个字符的权和。并将26个字母出现的位置依次保存到vector中。随后从0~len依次枚举每个位置的字符s[i]，并枚举它的下一个出现位置，再计算这段子串是否满足条件(1)。这种算法在第6组数据中就TLE了。无奈只能寻找新的解法。

由条件一可以推知sum[i]=sum[j-1]，如果扫描到第j位时先不加上权值，那么就有sum[i]==sum[j]了。如果我们把sum值放入s[i]的map中来统计以s[i]结尾时sum的出现次数，那么自然也满足了条件(2)。因此新的算法就不难想出了：事先开一个26位长的map，保存以该字符结尾的子串遇到的所有sum值的次数。从头到尾扫描一遍s，先令所有的sum值出现次数为0，每次先累加以s[i]结尾且权值之和恰好是sum(sum暂时不累加s[i]处的权值)的子串的个数，即m[s[i]][sum]，然后再更新s[i]处结尾的sum值，进而更新m[s[i]][sum]的值。详细细节见代码注释。

3.代码：

#define _CRT_SECURE_NO_WARNINGS 
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

#define N 100000+10
int word[26];
char s[N];
int main()
{
	//freopen("test.txt", "r", stdin);
	long long cnt = 0,sum=0;
	for (int i = 0; i < 26; i++)
		scanf("%d", word + i);
	scanf("%s", s);
	int len = strlen(s);
	map<long long, long long>m[26];
	for (int i = 0; i < len; i++)
	{
		int x = s[i] - 'a';//s[i]的序号
		cnt += m[x][sum];//先累加以s[i]结尾且权值之和恰好是sum的子串的个数
		sum += word[x];//更新sum值，累加上s[i]的权值
		m[x][sum]++;//更新sum值出现的次数
	}
	cout << cnt << endl;
	return 0;
}