蓝桥 挑选子串

#include <bits/stdc++.h>
using namespace std;
const int N=2e3+5;
int a[N];
int main()
{
  int n,m,k;
  cin >> n >> m >> k;
  for(int i=1;i<=n;i++){
    cin >> a[i];
  }
  int sum=0;
  int ans=0;
  for(int i=1,j=0;i<=n;i++){
      while(j<i||(sum<k&&j+1<=n)){
        sum=sum+(a[++j]>=m);
      }
      if(sum >=k){
        ans=ans+n-j+1;
        sum=sum-(a[i]>=m);
      }
  }
  cout << ans;
  return 0;
}

思路，使用快慢指针。

精髓（a[i]>=)是一个判断语句，当a[i]>=m后值为1，若不是值为0；