acwing.4960字串简写(双指针加前缀和)

输入样例：

4
abababdb a b

输出样例：

6

样例解释

符合条件的子串如下所示，中括号内是该子串

[abab]abdb
[ababab]db
[abababdb]
ab[abab]db
ab[ababdb]
abab[abdb]

 分析：利用两个指针从前往后分别寻找c1和c2，统计前一个c1的个数，后面每次寻找到一个c2可以和前面所有的c1组成一个字串

 

#include<bits/stdc++.h>
using namespace std;

typedef long long  ll;
ll res=0;
ll cnt=0;
string a;//字符串 
int k;
char c1,c2;
int main()
{
	cin>>k;
	cin>>a>>c1>>c2;
	
	for(int i=0,j=i+k-1;j<a.size();i++,j++)
	{
		if(a[i]==c1)//记录前面所有的c1 
		cnt++;
		if(a[j]==c2)
		res+=cnt;
	}
	cout<<res<<endl;
	
	return 0;
}