将不小于m的数看作1,剩下的数看作0,那么只要区间内1的个数不小于k则可行,枚举左端点,右端点可以通过two-pointer求出。
时间复杂度O(n)O(n)。
#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;
int a[200005],sum[200005];
int n,m,k;
int t;
int main()
{
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&k);
int x;
sum[0]=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
if(x>=m)
a[i]=1;
else
a[i]=0;
sum[i]=sum[i-1]+a[i];
}
long long ans=0;
int r=1;
for(int l=1;l<=n;l++)
{
while(r<=n&&sum[r]-sum[l-1]<k)
r++;
if(r>n)
break;
ans+=(n-r+1);
}
cout<<ans<<endl;
}
return 0;
}