题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=5806
作为HDU集训队的一员,我很清楚Claris是一位数据结构大师,但正是如此,比赛的时候小编老是往数据结构上去想,哎,然而此题完全不用(=.= GG),只需要用到双指针就能够解决。
仔细思考,我们是要区间里面第k大的数大于m,那么其实只要区间里面有k个大于m的数就可以了,接下来就只需要用到双指针就行了。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 200000+5;
typedef long long LL;
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=0; i<n; i++)
{
int tmp;
scanf("%d", &tmp);
a[i] = (tmp >= m);
}
LL L,R,sum,ans;
L = R = sum = ans = 0;
sum = a[0];
while(R < n)
{
if(sum == k)
{
ans += n-R;
sum -= a[L];
L++;
}
else
{
R++;
sum += a[R];
}
}
printf("%I64d\n", ans);
}
}