采用存储目标字符下标的方法,此题的想法比较新奇,故予以记录。
存好下标之后,可以先定位好启始的字符,然后去搜结尾字符符合长度k并且最靠近启始字符的下标,找到之后可以直接取到这个下标之后的所有下标,因为都满足了长度大于等于k。
代码:
#include<iostream>
#include<vector>
using namespace std;
const int N = 5e5 + 10;
int main() {
//读入
int k; cin >> k;
string str; cin >> str;
char c1, c2; cin >> c1 >> c2;
//a1存是字符c1的下标,a2存是字符c2的下标
vector<int>a1;
vector<int>a2;
//存
for (int i = 0; i < str.length(); i++) {
if (str[i] == c1)a1.push_back(i);
if (str[i] == c2)a2.push_back(i);
}
long long res = 0;//开longlong存答案
for (int i = 0; i < a1.size(); i++) {//扫所有的c1对应的下标
int t = a1[i]; //先取出c1对应下标
int c = t + k - 1; //定义出满足和c1距离k的下标
//二分,从0二分到存c2下标的数组的长度
int l = 0, r = a2.size() - 1;
while (l < r) {
int mid = l + r >> 1;
if (a2[mid] >= c)r = mid; //如果当前c2的下标是大于等于最近的位置,就去搜更小的
else l = mid + 1;
}
//这里直接取a2的长度减去l,l在这里代表的是a2的下标
//通过上面的二分,l已经是最靠近c1的了,所以这就是能够取到的最长的长度了
if (a2[l] >= c)res += a2.size() - l;
}
cout << res;
return 0;
}