CodeForces 19D A and B and Interesting Substrings 【前缀和】

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

题意分析：给定每个小写字母所对应的兴趣值，再给定一个字符串，求这个字符串中满足以下条件的子串的个数：长度大于等于2、首末字母相同、除首尾外中间字母兴趣值的和为0

解题思路：本题中涉及到区间和的问题。在每个数不再更改的情况下，一般直接记录每个位置的前缀和，就可以将区间和的问题转换为前缀和。这个题中，要求区间和为0且首尾相同的子串，我们只需求得当前字母之前相同字母，并且这些字母前缀和等于当前字母前缀和减去自己的兴趣值，这样的字母的个数就是以当前字母为末端的题中要求子串数。而要记录某一个数字出现多少次，很容易想到用map实现。

AC代码：

//CodeForces-19D A and B and Interesting Substrings
//AC 2016-4-20 22:02:32
//trick prefix-sum
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cctype>
#include <cstdlib>
#include <cstring>
#include <vector>
#include <set>
#include <string>
#include <map>
#include <queue>
#include <deque>
#include <list>
#include <sstream>
#include <stack>
using namespace std;

#define cls(x) memset(x,0,sizeof x)
#define inf(x) memset(x,0x3f,sizeof x)
#define neg(x) memset(x,-1,sizeof x)
#define ninf(x) memset(x,0xc0,sizeof x)
#define st0(x) memset(x,false,sizeof x)
#define st1(x) memset(x,true,sizeof x)
#define INF 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define abs(x) (x>0?x:-x)
#define max(x,y) (x>y?x:y)
#define min(x,y) (x<y?x:y)
#define bug cout<<"here"<<endl;
//#define debug

int val[26];
map<long long,long long> prefix[26];
char s[100007];

int main()
{
    #ifdef debug
        freopen("E:\\Documents\\code\\input.txt","r",stdin);
        freopen("E:\\Documents\\code\\output.txt","w",stdout);
    #endif
    for(int i=0;i<26;++i)
        scanf("%d",&val[i]);
    long long cur=0;
    long long ans=0;
    char c=0;
    getchar();
    while((c=getchar())!='\n')
    {
        c-='a';
        if(prefix[c].count(cur))
            ans+=prefix[c][cur];
        cur+=val[c];
        prefix[c][cur]++;
    }
    cout<<ans<<endl;
    return 0;
}

转载于:https://www.cnblogs.com/DrCarlluo/p/6580619.html