现在的位置: 首页 > 综合 > 正文

CF(#294 div2):D. A and B and Interesting Substrings

2019年09月03日 ⁄ 综合 ⁄ 共 1100字 ⁄ 字号 评论关闭

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

 对于此题,刚开始我想到的是先把每个位置的和求出来,用sum[i]保存,然后把每种字母所在的位置求出来,用vector[i]来保存。这样一来我们就可以枚举每种字母,在枚举这种字母的位置,求出满足和为零的区间就可以了。可没想到的是这样的时间达到了O(n^2),行不通!
看了别人的代码,恍然大悟:因为区间是没有方向性的,我们可以按一定方向找区间,可以从左到右,也可以从右到左。由于习惯性,就按从左到右吧!
这一样来,类似于Dp的以某点为起始点或以某点为终点的思想(这里选择以某点为终点),只需一次扫描即可。由于需要用map的查找操作为O(log n)所以总时间复杂度为O(nlog n)。
 
 
代码及详解:
 
 
//freopen("C:\\Documents and Settings\\All Users\\桌面\\in.txt","w",stdout);

#include<iostream>
#include<iostream>
#include<string>
#include<cstdio>
#include<vector>
#include<map>
#include<string>
#define LL __int64

using namespace std;

string s;
map<LL,LL> mp[30];//mp[i][j]表示字母i的和为j的个数
int val[30];
LL sum[100010];
vector<int> v[30];

int main(){
    //freopen("C:\\Documents and Settings\\All Users\\桌面\\kk.txt","r",stdin);
    while(cin>>val[0]){
        for(int i=0;i<30;i++){
            v[i].clear();
            mp[i].clear();
        }
        for(int i=1;i<26;i++)
            cin>>val[i];
        cin>>s;
        LL ans=0;
        for(int i=0;i<s.size();i++){
            int tmp=s[i]-'a';
            if(i==0){sum[i]=val[tmp];mp[tmp][sum[i]]++;continue;}
            if(mp[tmp].count(sum[i-1])) ans+=mp[tmp][sum[i-1]];
            sum[i]=sum[i-1]+val[tmp];
            mp[tmp][sum[i]]++;//记录在i的前面,tmp有多少个和为:sum[i]
        }
        printf("%I64d\n",ans);
    }
    return 0;
}

 
 
 

抱歉!评论已关闭.