记录一个菜逼的成长。。
将不小于m的数看作1,剩下的数看作0,那么只要区间内1的个数不小于k则可行,枚举左端点,右端点可以通过two-pointer求出。
时间复杂度O(n)。
/(ㄒoㄒ)/~~
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <list>
#include <deque>
#include <cctype>
#include <bitset>
#include <cmath>
using namespace std;
#define cl(a) memset(a,0,sizeof(a))
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
const int INF = 0x3f3f3f3f;
const int maxn = 200000 + 10;
int a[maxn];
int main()
{
int T;
scanf("%d",&T);
while(T--){
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for( int i = 1; i <= n; i++ ){
scanf("%d",a+i);
}
LL ans = 0;
int r = 1,num = 0;
for( int i = 1; i <= n; i++ ){
while(num < k && r <= n)num += (a[r++] >= m);//寻找右端点
if(num < k)break;//首先必须要有k个数
ans += (LL)(n - r + 2);
num -= (a[i] >= m);
}
printf("%lld\n",ans);
}
return 0;
}