HDU-5806 NanoApe Loves Sequence Ⅱ(尺取法)

题意

给定 n 个数,求第 k 大的数不小于 m 的区间的个数。
1 n 200000

思路

这道题的单调性是好找的,找到一个满足条件的区间,右端点更右的区间都满足条件。对于一个区间实时维护一个数大小的排行确实不方便,但我们可以适当转化题目。第 k 大的数大于等于 m ,相当于大于等于 m 的数的数量大于等于 k 。由此直接尺取区间即可。不难看出,尺取方便维护的是一个或多个数量的问题,可以做到实时保存修改。这种转化法有很强的拓展性。

代码

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define FOR(i,x,y) for(int i=(x);i<=(y);i++)
#define DOR(i,x,y) for(int i=(x);i>=(y);i--)
typedef long long LL;
using namespace std;
int a[200003];

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m,k,R=0,cnt=0;
        LL ans=0;
        scanf("%d%d%d",&n,&m,&k);
        FOR(i,1,n)scanf("%d",&a[i]);
        FOR(L,1,n)
        {
            while(R<=n&&cnt<k)cnt+=a[++R]>=m;
            if(R>n)break;
            ans+=n-R+1;
            cnt-=a[L]>=m;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/paulliant/article/details/80873843