NEU 1666 字符串游戏（巧用map）

**

1666: 字符串游戏

**

时间限制: 1 Sec 内存限制: 128 MB

题目描述

说到游戏，大家还是比较喜欢的，但是玩游戏，如果想赢得对方还是得靠思维的，xy比较喜欢字符串，尤其是起始字母等于终止字母这样的字符串(the string’s length must exceed one)，但是呢，xy有个癖好，喜欢把每个字符重新分配一个值，喜欢的字符给非负数，不喜欢的字符给负数。但是，xy比较喜欢数字0，所以，他决定找到一个字符串中有多少个字串满足起始字母等于终止字母，且除了起始字母和终止字母外其他字母的和为0。但是xy最近比较忙，希望你能帮助他。

输入

第一行包含26个整数xi，xi属于[-100000,100000]，非别代表着小写字母a,b,c,….,z的值。

第二行一个只包含小写字母的字符串，长度你自己猜，算了，给你个范围，length between 1 and 100000.

输出

输出答案，每个答案占一行。

样例输入

1 1 -1 1 1 1 1 1 1 1 1 1 1 1 1 7 1 1 1 8 1 1 1 1 1 1

xabcab

样例输出

2

用结构体node同时记录一下当前位置的前缀和以及当先位置的字符，再用map映射出出现的数量。因为map内部会用到比较，需要重载一下小于号。

对于一个位置，查找这个位置的字符和前一位置的前缀和所对应的node在map中的数量，ans加上这个数量就好。

ps：由于数据范围较大，为了防止在运算过程中爆int，全用long long 就好。

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <iostream>
#include <stack>
#include <map>
using namespace std;

struct Node
{
char c;
long long sum;
bool operator < (const Node& v) const
{
if (c!=v.c) return c<v.c;
else return sum<v.sum;
}
};

long long num[26];
map<Node,int> k;
char in[100001];

int main()
{

while(~scanf("%lld",&num[0]))
{
k.clear();
long long sum=0,ans=0;
for(int i=1;i<26;i++) scanf("%lld",&num[i]);
scanf("%s",in);
int len=strlen(in);
for(int i=0;i<len;i++)
{
Node kk;
kk.c=in[i];
kk.sum=sum;
sum+=num[in[i]-'a'];
//cout<<kk.c<<' '<<kk.sum<<' '<<k[kk]<<endl;
if (k[kk]==0)
{
kk.sum=sum;
k[kk]++;
}
else
{
ans+=k[kk];
kk.sum=sum;
k[kk]++;
}
}
cout<<ans<<endl;
}
return 0;
}