题目描述
程序猿圈子里正在流行一种很新的简写方法:
对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符用这部分的长度代替。
例如 internationalization 简写成 i18n,Kubernetes 简写成 K8s,Lanqiao 简写成 L5o 等。
在本题中,我们规定长度大于等于 K
的字符串都可以采用这种简写方法(长度小于 K
的字符串不配使用这种简写)。
给定一个字符串 S
和两个字符 c1
和 c2
,请你计算 S
有多少个以 c1
开头 c2
结尾的子串可以采用这种简写?
输入格式
第一行包含一个整数 K
。
第二行包含一个字符串 S
和两个字符 c1
和 c2
。
输出格式
一个整数代表答案。
数据范围
对于 20%
的数据,2≤K≤|S|≤10000
。
对于 100%
的数据,2≤K≤|S|≤5×105
。S
只包含小写字母。c1
和 c2
都是小写字母。
|S|
代表字符串 S
的长度。
输入样例:
4
abababdb a b
输出样例:
6
样例解释
符合条件的子串如下所示,中括号内是该子串
[abab]abdb
[ababab]db
[abababdb]
ab[abab]db
ab[ababdb]
abab[abdb]
算法1
(暴力枚举) $O(n^2)$
blablabla
时间复杂度o(n)
C++ 代码
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
int k,cnt;
string s;
char a,b;
int main()
{
cin>>k>>s>>a>>b;
int n=s.size();
ll ans=0; //答案记得开longlong
// for(int i=0,j=k-1;j<n;i++,j++)
// {
// if(s[i]==a)cnt++;
// if(s[j]==b)ans+=cnt;
// }
//为了方便理解写成后缀和形式
for(int i=n-k,j=n-1;i>=0;i--,j--)
{
if(s[j]==b)cnt++;
if(s[i]==a)ans+=cnt;
}
cout<<ans;
return 0;
}