NanoApe Loves Sequence Ⅱ

最新推荐文章于 2022-02-20 00:17:47 发布

原创最新推荐文章于 2022-02-20 00:17:47 发布 · 335 阅读

0 ·

CC 4.0 BY-SA版权

组合数学专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一个区间计数问题的解决方案，通过遍历数组并利用滑动窗口思想来统计大于特定阈值的元素数量，以此判断区间内符合条件的元素是否超过预设的最小数量k。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

将小于m的数为0，其余数为1，只要区间中1的个数大于k即可

#include <iostream>  
#include <cstdio>  
#include <cstring>  
#include <algorithm>  
#include <cmath>  
#include <queue>  
using namespace std;  
#define N 200050  
int a[N];  
int main()  
{  
    int T,n,m,k;  
    scanf("%d",&T);  
    while(T--)  
    {  
        scanf("%d %d %d",&n,&m,&k);  
        int t=0;  
        long long sum=0;  
        for(int i=1; i<=n; i++)  
            scanf("%d",&a[i]);  
        int r=1,num=0;  
        for(int i=1;i<=n;i++)  
        {  
            while(r<=n&&num<k)  
            {  
                if(a[r]>=m) num++;  
                r++;  
            }  
            if(num>=k) sum+=n-(r-1)+1;  
            if(a[i]>=m) num--;  
        }  
        printf("%lld\n",sum);  
    }  
    return 0;  
}