HDU 5806 NanoApe Loves Sequence Ⅱ（尺取）

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 222222int 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;}