(C++)蓝桥杯b组14届真题--字串简写

哈喽，大家好！！！又到了每日分享时刻，今天给大家带来的还是蓝桥杯的真题---字串简写。

题目描述

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

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

解题思路

第一次写，我是怎么简单怎么写，然而时间超限。主要代码如下：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int K,ans;
char c1,c2;
string S;
int main()
{
	cin>>K>>S>>c1>>c2;
	int x=S.size();
	for(int i=0,j=K-1;i<=x-K;i++,j++)
	{
		if(S[i]==c1)
		{
			for(int z=j;z<=x-1;z++)
			{
				if(S[z]==c2) ans++;
			}
		}
	}
	cout<<ans;
 } 

运行效果：

失败的原因是代码的复杂度太高，O(n^2),题目给的数据太大10^5。下面是优化的代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int K,sum;
ll ans;
char c1,c2;
string S;
int main()
{
	cin>>K>>S>>c1>>c2;
	int x=S.size();
	for(int i=0,j=K-1;i<=x-K;i++,j++)
	{
		if(S[i]==c1) sum++;
		if(S[j]==c2) ans+=sum;
	}
	cout<<ans;
	return 0;
 } 

代码运行效果如下：

看了b站上的一位优秀的博主，用的是前缀和的方法。大家可以自己写一个例子，带入代码中，你就会明白。

好啦！！！今天的分享就到这，喜欢的记得点赞+收藏+关注奥。祝大家每天开心😀😀😀，天天进步。