题目描述
程序猿圈子里正在流行一种很新的简写方法:
对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符用这部分的长度代替。
例如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]
#include<iostream>
#include<string>
using namespace std;
int main() {
int K;
string S;
char c1, c2;
int n1Num = 0;
long long ans = 0;
cin >> K;
cin >> S >> c1 >> c2;
for (int i = K - 1; i < S.size(); i ++) {
if (S[i - K + 1] == c1) {
n1Num++;
}
if (S[i] == c2) {
ans += n1Num;
}
}
cout << ans;
return 0;
}
补充:
最后结果比较大, 得用long long存储
两次循环就超时了,所以需要在一次循环中解决
可以先从K-1开始,然后记录之前c1出现的次数,当遇到c2时,就只需要将c1出现的次数加上去就可以了
可以看如下示例:
[abab]abdb b的位置为3 1个a
[ababab]db b的位置为5 2个a
[abababdb] b的位置为7 3个3
ab[abab]db b的位置为5
ab[ababdb] b的位置为7
abab[abdb] b的位置为7
所以就可以只需要遍历一次,就是只需要看b之前的大于等于K距离的a的个数