第十四届蓝桥杯省赛C++子串简写

程序猿圈子里正在流行一种很新的简写方法：
对于一个字符串，只保留首尾字符，将首尾字符之间的所有字符用这部分的长度代替。
例如 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]

难度：中等
时/空限制：1s / 256MB
总通过数：1419
总尝试数：4365
来源：
第十四届蓝桥杯省赛C++ B组

#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=5e5+10;

int n, k;
char a, b, s[N];

int main() {
    cin >> k >> s + 1 >> a >> b;

    n = strlen(s + 1);
    ll ans = 0;
    int cnt = 0;
    for (int i = k; i <= n; i ++ ) {
        if (s[i - k + 1] == a) cnt ++ ;
        if (s[i] == b) ans += cnt;
    }

    cout << ans;

    return 0;
}