题目描述
程序猿圈子里正在流行一种很新的简写方法:对于一个字符串,只保留首尾字符,将首尾字符之间的所有字符用这部分的长度代替。例如 internation-alization 简写成 i18n,Kubernetes (注意连字符不是字符串的一部分)简写成 K8s, Lanqiao 简写成 L5o 等。
在本题中,我们规定长度大于等于 K 的字符串都可以采用这种简写方法(长度小于 K 的字符串不配使用这种简写)。
给定一个字符串 S 和两个字符 c1 和 c2,请你计算 S 有多少个以 c1 开头c2 结尾的子串可以采用这种简写?
输入格式
第一行包含一个整数 K。
第二行包含一个字符串 S 和两个字符 c1 和 c2。
输出格式
一个整数代表答案。
样例输入
复制
4 abababdb a b
样例输出
复制
6
提示
符合条件的子串如下所示,中括号内是该子串:
[abab]abdb
[ababab]db
[abababdb]
ab[abab]db
ab[ababdb]
abab[abdb]
对于 20% 的数据,2 ≤ K ≤ |S | ≤ 10000。
对于 100% 的数据,2 ≤ K ≤ |S | ≤ 5 × 105。S 只包含小写字母。c1 和 c2 都是小写字母。
|S | 代表字符串 S 的长度。
个人分析
本题我最先想到的是暴力求解 利用两个for循环 一个遍历头一个遍历尾 很显然是会超时的
然后我使用的是一种前缀和的数学思想来求解这道题目
例如:
k = 4 start = a end = b
abababdb
利用 for (int i = 0; i < s.size(); i++)
s[i]在遍历头 s[j]在遍历尾
if (i > 0)
{
sum[i] = sum[i - 1];
}
if (s[i] == start)
{
sum[i]++;
}
sum[i] 代表的是从0开始到i位置 一共有多少个start字符
if (s[i] == end && i >= k - 1)
{
ans = ans + sum[i - k + 1];
}
这个就是计算在end字符的前面k个距离以前有多少个字符start
题解代码
#include<iostream>
using namespace std;
#include<string>
#include<vector>
int main()
{
int k;
cin >> k;
string s;
char start, end;
long long int ans = 0;
cin >> s >> start >> end;
vector<int> sum(s.size());
for (int i = 0; i < s.size(); i++)
{
if (i > 0)
{
sum[i] = sum[i - 1];
}
if (s[i] == start)
{
sum[i]++;
}
if (s[i] == end && i >= k - 1)
{
ans = ans + sum[i - k + 1];
//在距离s[j]的k位置前有多少个start
}
}
cout << ans << endl;
return 0;
}