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;    }}