题目描述

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

输入格式

第一行包含一个整数 K。
第二行包含一个字符串 S 和两个字符 c1 和 c2。

输出格式

一个整数代表答案。

样例输入

4
abababdb a b

样例输出

6

提示

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

评测用例规模与约定

对于 20% 的数据，2 ≤ K ≤ |S | ≤ 10000。
对于 100% 的数据，2 ≤ K ≤ |S | ≤ 5 × 10⁵。S 只包含小写字母。c1 和 c2 都是小写字母。
|S | 代表字符串 S 的长度。

前缀和

以c2结尾c1开头的子串数量为c2前的c1数量
子串总数量即为所有以c2结尾c1开头的子串数量之和
根据题目要求只统计长度大于k的子串
所以遍历c2时应统计位于c2-k+1的c1数量

C++ 代码

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int k;
    long long ans = 0, cnt = 0;
    string s;
    char c1, c2;
    cin >> k >> s >> c1 >> c2;
    // i从第一个可能的c1开始，j从第一个可能的c2开始
    for (int i = 0, j = k - 1; j < s.size(); i++,j++)
    {
        // c1的数量
        if (s[i] == c1)
            cnt++;
        // 以c2结尾的子串数量
        if (s[j] == c2)
            ans += cnt;
    }
    cout << ans;
    return 0;
}