程序猿圈子里正在流行一种很新的简写方法:
对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符用这部分的长度代替。
例如 internationalization
简写成 i18n
,Kubernetes
简写成 K8s
,Lanqiao
简写成 L5o
等。
在本题中,我们规定长度大于等于 K� 的字符串都可以采用这种简写方法(长度小于 K� 的字符串不配使用这种简写)。
给定一个字符串 S� 和两个字符 c1�1 和 c2�2,请你计算 S� 有多少个以 c1�1 开头 c2�2 结尾的子串可以采用这种简写?
输入格式
第一行包含一个整数 K�。
第二行包含一个字符串 S� 和两个字符 c1�1 和 c2�2。
输出格式
一个整数代表答案。
数据范围
对于 20%20% 的数据,2≤K≤|S|≤100002≤�≤|�|≤10000。
对于 100%100% 的数据,2≤K≤|S|≤5×1052≤�≤|�|≤5×105。S� 只包含小写字母。c1�1 和 c2�2 都是小写字母。
|S||�| 代表字符串 S� 的长度。
输入样例:
4
abababdb a b
输出样例:
6
样例解释
符合条件的子串如下所示,中括号内是该子串
[abab]abdb
[ababab]db
[abababdb]
ab[abab]db
ab[ababdb]
abab[abdb]
思路:
#include<bits/stdc++.h>
using namespace std;
int main()
{
int k;
long long cnt=0,sum=0;
char c1,c2;
string s1;
cin>>k;
cin>>s1>>c1>>c2;
int n=s1.size();
for(int i=0,j=k-1;j<n;i++,j++) //固定数量为k的区间,从0开始移动
{
if(s1[i]==c1)cnt++; //记录c1的个数
if(s1[j]==c2)sum+=cnt; //每一个c2与前面的所有c1构成的子串都符合条件,因此出现一个c2就要加上前面c1的个数
}
printf("%lld",sum);
return 0;
}//这里填你的代码^^
作者:yizhima
链接:https://www.acwing.com/activity/content/code/content/8346552/
来源:AcWing
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。