（尺取法）HDU - 5806 NanoApe Loves Sequence Ⅱ

最新推荐文章于 2019-05-18 14:09:15 发布

原创最新推荐文章于 2019-05-18 14:09:15 发布 · 336 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

ACM-hdu

ACM-技巧

本文介绍了一道经典的尺取法题目，通过寻找区间内第k大的值大于等于m的所有可能区间数量，详细解析了解题思路及代码实现。

原题链接：

http://acm.hdu.edu.cn/showproblem.php?pid=5806

题意：

给你一个m和k，求在n个元素的数列里有多少个区间的第k大的值大于等于m。

分析：

又是区间题，但是这题一看就感觉是尺取法。我们可以发现一个规律，如果一个区间存在第k大大于等于m，那每次加进新的值，也就是扩大区间的时候，如果后一个值小于等于原来的第k大，那么第k大不变。如果插入的数比原来的第k大大，那么新的第k大肯定比原来的第k大大，肯定满足条件，所以说只要枚举每一个起点，找到第一次满足第k大的值大于等于m，后面的剩余的区间都满足。但是感觉不知道怎么跑到第一次满足条件，坑了好久。

无意中听到学长的讨论，才发现自己还是想复杂了，确切的说是思维定势了，其实可以转个弯，要求有多少个区间的第k大值大于等于m，第一次满足条件只需要考虑，存在k个值大于等于m。
其他都无需考虑。这样就变成了一个非常标准的尺取法。代码非常短。

代码：

#include <iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<algorithm>
#include<string>
#include<queue>
#include<cmath>
#include<stack>
#include<cctype>
#include<list>


#define ll long long
#define ull unsigned long long

using namespace std;

const int maxn=200010;
const int inf=1<<30;

ll num[maxn];


int main()
{
    //#define DEBUG

#ifdef DEBUG
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
#endif

    int t;
    scanf("%d",&t);
    while(t--){
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=0;i<n;i++){
            scanf("%I64d",&num[i]);
        }
        ll ans=0,res=0;
        int l=0,r=0;
        if(num[0]>=m)res++;
        for(;r<n;){
            while(res<k){
                r++;
                if(r>=n)break;
                if(num[r]>=m){
                    res++;
                }
            }
            while(res>=k){
                ans+=n-r;
                if(num[l]>=m){
                    res--;
                }
                l++;
            }
        }
        printf("%I64d\n",ans);
    }
    return 0;
}