子串简写（蓝桥杯）

题目链接

​​​​蓝桥杯2023年第十四届省赛真题-子串简写 - C语言网

题目描述

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

|S | 代表字符串 S 的长度。

思路

遍历字符串，遇到c1时，记录c1出现的次数，遇到c2时，从可以简写的位置开始找到c1的位置，结果加上储存的c1个数，遍历完后即是答案

代码

#include <bits/stdc++.h> // 万能头文件
#define int long long // int转化为long long
using namespace std;

string s;
const int N = 5e5 + 10;
int a[N];

signed main() // 与int一样
{
    ios::sync_with_stdio(0); // 提高cin，cout速率
    cin.tie(0), cout.tie(0);
    int k, sum = 0, count = 1;
    char c1, c2;
    cin >> k >> s >> c1 >> c2;
    int n = s.size();
    for (int i = 0; i < n; ++i)
    {
        if (s[i] == c1)
            a[i] = count++;
        if (s[i] == c2)
        {
            for (int j = i - k + 1; j >= 0; --j)
            {
                if (a[j] != 0)
                {
                    sum += a[j];
                    break;
                }
            }
        }
    }
    cout << sum << '\n';
    return 0;
}