2023年第14届蓝桥杯大赛软件赛省赛 C/C++ 大学 B 组 试题 G: 子串简写（20）_十四届蓝桥杯省赛真题

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

参考代码:

#include <bits/stdc++.h>
using namespace std;
int main(void) {
    string s;
    char a, b;
    int k;
    long long ans = 0, cnt = 0;
    cin >> k;
    cin >> s >> a >> b;
    int len = s.length();
    for (int i = 0; i < len; i++) {
        if (i - k + 1 >= 0 && s[i - k + 1] == a) cnt++;
        if (s[i] == b) ans += cnt;
    }
    printf("%lld\n", ans);
    return 0;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19