【CCPC2017哈尔滨B】K-th Number

原题：

 

 

求f(x)不好求，那么可以尝试求f(x>=m)

求x为第k大出现了几次不好求，那么可以尝试求大于等于x的所有数作为第k大出现了几次

那么可以二分x，然后把所有大于等于x的视为1，其他视为0

问题转化为求有多少个区间，使得区间内1的个数>=k

区间问题可以枚举左端点，然后考虑右端点

显然右端点是递增的，所以不用二分，直接双指针扫一遍即可

本题本质在于用第一句话提到的方法，模糊掉不同的数，使得统计变得容易

 

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define LL long long
int n,a[110000],o;  LL m;
int nx,nn;
//LL s[110000];
LL cclt(int x){
    //for(int i=1;i<=n;++i)  s[i]=s[i-1]+(a[i]>=x);
    int j=0,bwl=0;  LL cnt=0;
    for(int i=1;i<=n;++i){
        bwl-=(a[i-1]>=x);
        while(bwl<o && j<n)  bwl+=(a[++j]>=x);
        if(bwl<o)  break;
        cnt+=(n-j+1);
    }
    return cnt;
}
bool chck(int x){
    return cclt(x) >= m;
}
int bnrsch(){
    int l=nn,r=nx,md;
    while(l+1<r){
        md=(l+r)>>1;
        (chck(md) ? l : r)=md;
    }
    return chck(r) ? r : l;
}
void prvs(){
    nx=0,nn=1000000007;
}
int main(){
    int T;  cin>>T;
    while(T --> 0){
        scanf("%d%d%lld",&n,&o,&m);
        prvs();
        for(int i=1;i<=n;++i){
            scanf("%d",&a[i]);
            nx=max(nx,a[i]);
            nn=min(nn,a[i]);
        }
        printf("%d\n",bnrsch());
    }
    return 0;
}