NanoApe Loves Sequence Ⅱ(尺取法）

NanoApe Loves Sequence Ⅱ

 

 Accepts: 374

 

 Submissions: 946

 Time Limit: 4000/2000 MS (Java/Others)

 

 Memory Limit: 262144/131072 K (Java/Others)

问题描述

退役狗 NanoApe 滚回去学文化课啦！

在数学课上，NanoApe 心痒痒又玩起了数列。他在纸上随便写了一个长度为 $n$ 的数列，他又根据心情写下了一个数 $m$。

他想知道这个数列中有多少个区间里的第 $k$ 大的数不小于 $m$，当然首先这个区间必须至少要有 $k$ 个数啦。

输入描述

第一行为一个正整数 $T$，表示数据组数。

每组数据的第一行为三个整数 $n,m,k$。

第二行为 $n$ 个整数 AiA_iA​i​​，表示这个数列。

1≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,Ai≤1091 \le T \le 10,~2 \le n \le 200000,~1 \le k \le n/2,~1 \le m,A_i \le 10^91≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,A​i​​≤10​9​​

输出描述

对于每组数据输出一行一个数表示答案。

输入样例

1
7 4 2
4 2 7 7 6 5 1

输出样例

18

const int N = 200005;
int t, n, m, k;
int a[N];
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 j = 1, i = 1, size = 0;
        while (i <= n && j <= n)
        {
            if (size < k) 
            {
                if (a[j] >= m) size++;
                j++;
            }
            while (size == k)
            {
                ans += n - (j-1) + 1;
                if (a[i] >= m) size--;
                i++;
            }
        }
        printf("%I64d\n",ans);
    }
}