hdu5806 two-pointers

来源:互联网 发布:nginx sub filter 编辑:程序博客网 时间:2024/06/11 20:36

头尾两个指针扫就可以了,注意ans非常大,要用long long ,输入输出要配套

#include <iostream>#include <cstdio>#include <cstring>#include <cstdlib>using namespace std;int T,n,m,k,x,s,i,j;long long ans;int a[200010];int main(){    scanf("%d", &T);    while (T--)    {        scanf("%d%d%d", &n, &m, &k);        memset(a,sizeof(a),0);        for (int i=1;i<=n;i++)        {            scanf("%d", &x);            if (x>=m) a[i]=1; else a[i]=0;        }        ans=0;        s=a[1];i=1;j=1;        while (i<=j && j<=n)        {            if (s>=k)            {                ans+=(n-j+1);                s-=a[i];                i++;            }            while (s<k && j<=n) {j++;s+=a[j];}        }        printf("%I64d\n", ans);    }}


0 0
原创粉丝点击