蓝桥杯2023年第十四届省赛真题-子串简写

题目描述

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

个人分析

本题我最先想到的是暴力求解 利用两个for循环 一个遍历头一个遍历尾 很显然是会超时的

然后我使用的是一种前缀和的数学思想来求解这道题目

例如：

k = 4  start = a  end = b

abababdb

利用 for (int i = 0; i < s.size(); i++)

s[i]在遍历头 s[j]在遍历尾 

        if (i > 0)
        {
            sum[i] = sum[i - 1];
        }
        if (s[i] == start)
        {
            sum[i]++;
        }

sum[i] 代表的是从0开始到i位置 一共有多少个start字符

        if (s[i] == end && i >= k - 1)
        {
            ans = ans + sum[i - k + 1];
        }

这个就是计算在end字符的前面k个距离以前有多少个字符start

题解代码

#include<iostream>
using namespace std;
#include<string>
#include<vector>

int main()
{
	 int k;
	cin >> k;
	string s;
	char start, end;
	long long int ans = 0;
	cin >> s >> start >> end;
	vector<int> sum(s.size());
	for (int i = 0; i < s.size(); i++)
	{
		if (i > 0)
		{
			sum[i] = sum[i - 1];
		}
		if (s[i] == start)
		{
			sum[i]++;
		}
		if (s[i] == end && i >= k - 1)
		{
			ans = ans + sum[i - k + 1];
			//在距离s[j]的k位置前有多少个start
		}
	}
	cout << ans << endl;


	return 0;
}