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

蓝桥杯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 的长度。

代码详解

思路

当时比赛时采用了两重循环进行暴力，显然只能得到部分分数，考完才发现这题能够使用动态规划中的线性dp进行解答，这部分内容可通过acwing中算法基础课进行复习解答
统计符合条件的字串，实质上就是在条件范围内遍历头部，然后找所有大于等于k的c2的数量

代码

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
typedef long long LL;//用long long 保证结果范围
int k;
string str;
char c1,c2;
int f[N];//表示以i开头c2的数量
LL res;//统计结果
int main(){
	 cin>>k;
	 cin>>str;
	 cin>>c1>>c2;
	 int len=str.length();
	 for(int i=len-1;i>=0;i--){
	 	f[i]=f[i+1];//逆序表示以i结尾才的数量首先等于以i+1结尾c2的数量
	 	if(str[i]==c2) f[i]++;//当str[i]的值为c2时此时数量+1
	 }
	 for(int i=0;i+k-1<len;i++){
	 	if(str[i]==c1) res+=f[i+k-1];//当str[i]的值为c1时，此时可简写的总数为大于等于k处所有c2数量的累加
	 	cout<<res<<endl;
	 }
	 cout<<res;
	 return 0;
}