HDU 5806

若 [i, j] 满足， 则 [i, j+1], [i, j+2]...[i,n]均满足

故设当前区间里个数为size， 对于每个 i ，找到刚满足 size == k 的 [i, j]， ans += n - j + 1 .

i++ 的时候看看需不需要size-- 就可以更新了。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define LL long long
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
",ans);
    }
}