【杭电oj】5806 - NanoApe Loves Sequence Ⅱ

最新推荐文章于 2020-06-26 23:18:32 发布

原创最新推荐文章于 2020-06-26 23:18:32 发布 · 211 阅读

0 ·

CC 4.0 BY-SA版权

杭电oj 同时被 2 个专栏收录

13 篇文章

订阅专栏

二分尺取

7 篇文章

订阅专栏

本文介绍了一个使用尺取法解决特定子串计数问题的算法实现。该问题要求统计长度为n的字符串中至少包含k个大于等于m的数的子串个数。通过将符合条件的元素标记为1，不符合条件的标记为0，问题转化为求含至少k个1的连续子串数目。

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

题目链接

题意是说统计一个长度为n的字符串中至少有k个大于等于m的数的字串的个数。
如果输入的时候就直接将大于等于m的元素标记为1，小于的标记为0，则本题就是求含有至少k的1的连续字串的个数。

我用的尺取法。

AC代码

#include<iostream>
#include<stdio.h>
using namespace std;

int a[210000];
int main(){
    int T,n,m,k;
    cin>>T;
    while(T--){
        cin>>n>>m>>k;
        for(int i=0;i<n;++i)    scanf("%d",&a[i]);
        int l = 0,r = 0,cnt = 0;
        long long ans = 0;
        while(r<n){
            if(a[r] >= m)
                cnt++;
            if(cnt == k){
                while(a[l] < m){
                    ans += n-r;
                    l++;
                }
                ans += n-r;
                cnt--;
                l++;
            }
            r++;
        }
        printf("%lld\n",ans);
    }
    return 0;
}