Description
退役狗 NanoApe 滚回去学文化课啦!
在数学课上,NanoApe 心痒痒又玩起了数列。他在纸上随便写了一个长度为n的数列,他又根据心情写下了一个数m。
他想知道这个数列中有多少个区间里的第k大的数不小于m,当然首先这个区间必须至少要有k个数啦。
Input
第一行为一个正整数T,表示数据组数。
每组数据的第一行为三个整数 n,m,k
第二行为n个整数Ai,表示这个数列。
1≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,Ai<=10^9
Output
对于每组数据输出一行一个数表示答案。
Sample Input
1
7 4 2
4 2 7 7 6 5 1
Sample Output
18
Solution
如果区间[l,r]中不小于m的数已经有k个的话,说明这个区间第k大的数不小于m,之后往这个区间中加数也不影响这个性质,故用尺取每次找到一个符合条件(即不小于m的数有k个)的[l,r]之后,直接往答案中累加n-r+1,然后l++即可,时间复杂度O(2n)
Code
#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
#define maxn 222222
int T,n,m,k,a[maxn];
ll ans;
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
ans=0;
int cnt=0,l=1,r=1;
while(l<=n)
{
while(r<=n&&cnt<k)
{
if(a[r]>=m)cnt++;
r++;
}
if(r==n+1&&cnt<k)break;
if(cnt==k)ans+=n-r+2;
if(a[l]>=m)cnt--;
l++;
}
printf("%I64d\n",ans);
}
return 0;
}