CodeForces 519D A and B and Interesting Substrings 前缀和

题意:给出26个小写字母的权值,给出一个字符串s,要求找出所有这样的子串,他们的首尾字符相同,并且除了首尾外其余字母首尾之和为0的子串。将数目输出出来。

思路:找前缀和相同,并且字符相同的数目。

简直是黑科技

#include<cstdio>
#include<queue>
#include<iostream>
#include<vector>
#include<map>
#include<cstring>
#include<string>
#include<set>
#include<stack>
#include<algorithm>
#define cle(a) memset(a,0,sizeof(a))
#define inf(a) memset(a,ox3f,sizeof(a))
#define ll long long
#define Rep(i,a,n) for(int i=a;i<=n;i++)
using namespace std;
const int INF = ( 2e9 ) + 2;
const int maxn = 28;
const int maxm = 1e5+10;
map<ll,ll> mp[26];// 找前缀和相同的两个字符 
int a[maxn];
char s[maxm];
int main()
{
//	freopen("in.txt","r",stdin);
	for(int i=0;i<26;i++)scanf("%d",&a[i]);
	scanf("%s",s+1);
	int len=strlen(s+1);
	ll ans=0,pre=0;
	for(int i=1;i<=len;i++)
	{
		int m=s[i]-'a';
		ans+=mp[m][pre];
		pre+=a[m];
		mp[m][pre]++;
	}
	printf("%I64d\n",ans);
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值