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 
#include 
#include 
#include 
#include
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(rif(numif(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;iscanf("%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<

你可能感兴趣的:(尺取法)