题意
给定
个数,求第
大的数不小于
的区间的个数。
思路
这道题的单调性是好找的,找到一个满足条件的区间,右端点更右的区间都满足条件。对于一个区间实时维护一个数大小的排行确实不方便,但我们可以适当转化题目。第 大的数大于等于 ,相当于大于等于 的数的数量大于等于 。由此直接尺取区间即可。不难看出,尺取方便维护的是一个或多个数量的问题,可以做到实时保存修改。这种转化法有很强的拓展性。
代码
#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
typedef long long LL;
using namespace std;
int a[200003];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m,k,R=0,cnt=0;
LL ans=0;
scanf("%d%d%d",&n,&m,&k);
FOR(i,1,n)scanf("%d",&a[i]);
FOR(L,1,n)
{
while(R<=n&&cnt<k)cnt+=a[++R]>=m;
if(R>n)break;
ans+=n-R+1;
cnt-=a[L]>=m;
}
printf("%lld\n",ans);
}
return 0;
}