这道题我们的做法是开两个vector,分别把a和b字符的下标存进去,然后遍历a字符,我们要求长度必须大于等于k,我们可以画个图,也就是说b的下标减a的下标必须大于等于k-1 也就是b的下标必须大于等于a的下标+k-1 我们用二分找出b数组里面大于等于这个值最小的数的下标,然后用nb减去这个下标就是满足要求的个数了
#include <iostream>
#include <vector>
using namespace std;
typedef long long LL;
string s;
LL ret;
vector<LL> a, b;
char ca, cb;
LL k;
int main()
{
cin >> k >> s >> ca >> cb;
LL ns = s.size();
for (int i = 0; i < ns; i++)
{
if (s[i] == ca) a.push_back(i);
else if (s[i] == cb) b.push_back(i);
}
LL na = a.size();
LL nb = b.size();
for (int i = 0; i < na; i++)
{
LL val = a[i] + k - 1;
LL left = 0, right = nb - 1;
while (left < right)
{
LL mid = (left + right) / 2;
if (b[mid] >= val) right = mid;
else left = mid + 1;
}
if (b[left] >= val)
ret += nb- left;
}
cout << ret << endl;
}