HDU 5806 NanoApe Loves Sequence Ⅱ（尺取）

最新推荐文章于 2021-04-12 00:29:34 发布

v5zsq

最新推荐文章于 2021-04-12 00:29:34 发布

阅读量490

点赞数

CC 4.0 BY-SA版权

分类专栏： HDU 尺取

本文链接：https://blog.youkuaiyun.com/V5ZSQ/article/details/52150929

HDU 同时被 2 个专栏收录

803 篇文章

订阅专栏

尺取

19 篇文章

订阅专栏

探讨了如何通过尺取法解决特定数列问题，寻找满足条件的区间数量。

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

Description
退役狗 NanoApe 滚回去学文化课啦！
在数学课上，NanoApe 心痒痒又玩起了数列。他在纸上随便写了一个长度为n的数列，他又根据心情写下了一个数m。
他想知道这个数列中有多少个区间里的第k大的数不小于m，当然首先这个区间必须至少要有k个数啦。
Input
第一行为一个正整数T，表示数据组数。
每组数据的第一行为三个整数 n,m,k
第二行为n个整数Ai，表示这个数列。
1≤T≤10, 2≤n≤200000, 1≤k≤n/2, 1≤m,Ai<=10^9
Output
对于每组数据输出一行一个数表示答案。
Sample Input
1
7 4 2
4 2 7 7 6 5 1
Sample Output
18
Solution
如果区间[l,r]中不小于m的数已经有k个的话，说明这个区间第k大的数不小于m，之后往这个区间中加数也不影响这个性质，故用尺取每次找到一个符合条件（即不小于m的数有k个）的[l,r]之后，直接往答案中累加n-r+1，然后l++即可，时间复杂度O(2n)
Code

#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
#define maxn 222222
int T,n,m,k,a[maxn];
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 cnt=0,l=1,r=1;
        while(l<=n)
        {
            while(r<=n&&cnt<k)
            {
                if(a[r]>=m)cnt++;
                r++;
            }
            if(r==n+1&&cnt<k)break; 
            if(cnt==k)ans+=n-r+2;
            if(a[l]>=m)cnt--;
            l++;
        }
        printf("%I64d\n",ans);
    }
    return 0;
}