样例输入:
4
abababdb a b
样例输出:
6
分析:
这道题还是比较简单的,我们需要记录字符c1和c2出现的位置,对于每一个c2出现的位置,我们都可以寻找在其前面且距离大于等于k的字符1的出现次数,这个可以直接二分求解,也可以用前缀和进行优化,复杂度都是可以的。需要注意的一点是c1和c2有可能相同。
细节见代码:
#include<iostream>
#include<cmath>
#include<queue>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=5e5+10;
char s[N];
int p[N],tt;
int main()
{
int k;
scanf("%d",&k);
scanf("%s",s+1);
int n=strlen(s+1);
char c1[2],c2[2];
scanf("%s%s",c1,c2);
long long ans=0;
for(int i=1;i<=n;i++)
{
if(c1[0]==s[i]) p[++tt]=i;
if(c2[0]==s[i])
{
if(!tt||i-p[1]+1<k) continue;
int l=1,r=tt;
while(l<r)
{
int mid=l+r+1>>1;
if(i-p[mid]+1>=k) l=mid;
else r=mid-1;
}
ans+=l;
}
}
printf("%lld",ans);
return 0;
}