题目描述
程序猿圈子里正在流行一种很新的简写方法:
对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符这部分的长度代替。
例如 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
AC代码
//由于只考虑首尾字母,我们可以创建个辅助数组记录每个大于K字母后和C2一样的字母个数
//最后通过枚举每一个 大于等于K 的区间,将所有符合要求的串全部加起来即可
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int N = 500010;
int k;
char g[N],c1,c2;
ll num[N];
int main()
{
cin >> k >> g >> c1 >> c2;
int n = strlen(g);
for (int i = n - 1 ; i >= k - 1 ; i --)
{
if(g[i] == c2) num[i] = num[i + 1] + 1;
else num[i] = num[i + 1];
//cout << num[i] << " ";
}
ll cnt = 0;
for (int i = 0 ; i < n - k + 1; i ++)
{
int j = i + k - 1;
if(g[i] == c1) cnt += num[j];
}
printf("%lld",cnt);
return 0;
}