[蓝桥杯2023年第十四届省赛真题-子串简写] 一次循环方法

十年OI一场空 不开long long见祖宗

参考代码:

#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;

}