第十四届蓝桥杯.子串简写（前缀和\后缀和）

程序猿圈子里正在流行一种很新的简写方法：
对于一个字符串，只保留首尾字符，将首尾字符之间的所有字符用这部分的长度代替。
例如internationalization简写成 i18n，Kubernetes 简写成 K8s, Lanqiao 简写成 L5o 等。
在本题中，我们规定长度大于等于 K 的字符串都可以采用这种简写方法。
长度小于 K 的字符串不允许使用这种简写。
给定一个字符串 S 和两个字符 c1 和 c2。
请你计算 S 有多少个以 c1 开头 c2 结尾的子串可以采用这种简写？

输入格式

第一行包含一个整数 K。
第二行包含一个字符串 S 和两个字符c1 和c2。
对于 20% 的数据，2 ≤ K ≤ |S| ≤ 10000。
对于 100% 的数据，2 ≤ K ≤ |S| ≤ 5 × 105。
S 只包含小写字母。c1 和 c2 都是小写字母。
|S| 代表字符串S 的长度。

输出格式

一个整数代表答案。

输入样例

4
abababdb a b

输出样例

6

数据范围与提示

符合条件的子串如下所示，中括号内是该子串：
[abab]abdb
[ababab]db
[abababdb]
ab[abab]db
ab[ababdb]
abab[abdb]

解析：前缀和或后缀和都可以，其中后缀和可以省略判断是否越界，注意开long long

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
long long k,sum[N],res;
string s;
char c1,c2;
int main(){
	cin>>k>>s>>c1>>c2;
	for(int i=s.size()-1;i>=0;i--){
		if(s[i]==c2) sum[i]=sum[i+1]+1;
		else{
			sum[i]=sum[i+1];
			if(s[i]==c1) res+=sum[i+k-1];
		}
		if(c1==c2) res+=sum[i+k-1];		//防止c1==c2导致不执行11行else 
	}
	cout<<res;
    return 0;
}