字串简写
题目来源
蓝桥杯 2023 省赛
原题链接
问题描述
程序猿圈子里正在流行一种很新的简写方法：对于一个字符串，只保留首尾字符 ，将首尾字符之间的所有字符用这部分的长度代替。例如 internation-alization 简写成 i18n，Kubernetes （注意连字符不是字符串的一部分）简写成 K8s, Lanqiao 简写成 L5o 等。
在本题中，我们规定长度大于等于 K 的字符串都可以采用这种简写方法（长度小于 K 的字符串不配使用这种简写）。
给定一个字符串 S和两个字符 c1和 c 2 ，请你计算 S 有多少个以 c1开头 c 2结尾的子串可以采用这种简写？
输入格式
第一行包含一个整数 K。
第二行包含一个字符串 S 和两个字符 c 1 和 c 2。
输出格式
一个整数代表答案。
样例输入
4
abababdb a b
copy
样例输出
6
copy
完整代码
暴力算法
通过率70%
cpp
#include <iostream>
#include <cstring>
using namespace std;
int n; //要求字符串长度大于n
string s; //字符串
char a,b; //起始、终止字符
int main()
{
cin>>n;
cin>>s>>a>>b;
int res=0;
//遍历字符串
for(int i=0;i<=s.size()-n;i++){
//若找到与起始字符a相同的字符
if(s[i]==a){
int j=i+n-1; //长度大于n
//遍历剩余字符串
while(j<s.size()){
if(s[j]==b) res++;
j++;
}
}
}
cout<<res<<endl;
return 0;
}优化
通过率100%
cpp
#include <iostream>
#include <cstring>
using namespace std;
const int N=5e5+10;
int n; //要求字符串长度大于n
char a,b; //起始、终止字符
string s; //字符串
int sumb[N]; //记录从当前位置开始以后终止字符串的个数
int main()
{
cin>>n>>s>>a>>b;
long long res=0;
int last=0; //记录前一个终止字符记录的个数
//逆序遍历
for(int i=s.size()-1;i>=0;i--){
if(s[i]==b){
sumb[i]=last+1; //从当前位置i开始往后的终止字符b的个数=后一个出现的b记录的个数sumb[i]+当前的一个b
last=sumb[i];
}
}
for(int i=0;i<=s.size()-n;i++){
if(s[i]==a){
int j=i+n-1;
while(j!=s.size()){
if(sumb[j]!=0){
res+=sumb[j];
break;
}
else j++;
}
}
}
cout<<res<<endl;
return 0;
}