HDU 6231 K-th Number 尺取法

题目链接

题意： 给出一个数组a 将其所有子段中的第k大的数字放进b中，求b中的第m大的数字。
解题思路： 巧妙的取尺法。取尺法可以在O（n）的时间内很轻松的求出以大于等于x为第k大的区间有多少个。枚举每一个左端点l，使得整个区间内的大于等于x的数为k个，此时符合条件的区间数要加上（n-r）
因为当右端点在[r,n-1]这个区间内时，去k大值一定是大于等于x的。
想通了这个尺取法，就应该想到二分查找，因为这个是有单调性的。

#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include<iostream>
using namespace std;
typedef long long LL;
const int MAX = 1000000 + 100;
long long n,k,m;
int a[MAX];
int b[MAX];
bool check(int x){
    LL ans=0;
    int l=0,r=-1;
    int num=0;
    while(r<n){
        if(num<k){
            if(a[r+1]>=x) num++;
            r++;
        }else {
            if(num==k)ans+=(n-r);
            if(a[l]>=x) num--;
            l++;
            if(ans>=m) return true;
        }
    }
    return false;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%d %d %lld",&n,&k,&m);
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        int l=0,r=n-1;
        sort(b,b+n);
        int m;
        int ans=-1;
        while(l<=r){
            m=(l+r)>>1;
            if(check(b[m])){
                ans=b[m];
                l=m+1;
            }else{
                r=m-1;
            }
        }
        cout<<ans<<endl;
    }
}