题目描述

子串简写（二分）
将两个字符的位置分别记录下来 遍历第一个字符的位置
每次二分出它在第二个数组中满足条件的最小位置 把剩下的数量加到答案中
时间复杂度是 nlogn

C++ 代码

#include <iostream>
using namespace std;

typedef long long LL;
const int N = 5e5 + 10;

string s;
char c1,c2;
int n,m,k;
int a[N],b[N];

int main()
{
  cin>>k>>s>>c1>>c2;

  for(int i = 0;i < s.size();i ++)
  {
    if(s[i] == c1) a[n ++] = i;
    if(s[i] == c2) b[m ++] = i;
  }

  LL res = 0;
  for(int i = 0;i < n;i ++)
  {
    int l = 0,r = m;
    while(l < r)
    {
      int mid = (l + r) / 2;
      if(b[mid] >= a[i] + k - 1) r = mid;
      else l = mid + 1;
    }
    res += m - l;
  }
  cout<<res;
}