hdu5806尺取法

将不小于m的数看作1,剩下的数看作0,那么只要区间内1的个数不小于k则可行,枚举左端点,右端点可以通过two-pointer求出。

时间复杂度O(n)O(n)。

#include <cstdio>
#include <algorithm>
#include <iostream>
using namespace std;

int a[200005],sum[200005];
int n,m,k;
int t;
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&n,&m,&k);
        int x;
        sum[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&x);
            if(x>=m)
            a[i]=1;
            else
            a[i]=0;
            sum[i]=sum[i-1]+a[i];
        }
        long long ans=0;
        int r=1;
        for(int l=1;l<=n;l++)
        {
            while(r<=n&&sum[r]-sum[l-1]<k)
            r++;
            if(r>n)
            break;
            ans+=(n-r+1);
        }
        cout<<ans<<endl;
    }
    return 0;
}
    原文作者:B树
    原文地址: https://blog.csdn.net/aonaigayiximasi/article/details/52165956
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞