字串简写
题目来源
蓝桥杯 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%
#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%
#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]+当前的一个blast=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;
}