HDU 6231 (二分+尺取+烧脑)

所求的一定是a数组中的原素,这点毋庸置疑,所以在a数组中任意选则其中的一个数x,若a的所有长度不小于k的连续子序列中第k大数不小于x的子序列一共有ans个,那么x在所有第k大元素组成的数列中至少是第ans大数;(因为ans+=n-r,因为之前的x已经是第k大了,所以加上后面的元素,如果比他小,则没有影响,如果比他大,他就是第k大中的第二大,依次类推,知道加到n-r,此

时x就能保证至少是第ans大,用此时的ans跟m做比较进行二分查找)

感觉他大体的思路是这样,因为要找到第k大中的第m大,直接找肯定不可能的,而k,m又都是已知的,所以倒着看就是在数组中取出一个数看他是否能作为第k大中的第m大,为了快速选取,则进行二分查找操作,判断x是否能作为第k大,如果能,后面直接就不用考虑了,加上去至少ans大,所以就能比较出答案。确实有些烧脑。

#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

const int maxn=1e5+10;

int a[maxn],b[maxn];

int n,k;

typedef long long ll;

ll m;

ll calc(int x){
    ll ans=0;
    int l=0,r=-1,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++;
        }
    }
    return ans;
}

int main(){
    int cs;
    scanf("%d",&cs);
    while(cs--){
        memset(a,0,sizeof(a));
        scanf("%d%d%lld",&n,&k,&m);
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        sort(b,b+n);
        int len=unique(b,b+n)-b;
        int l=0,r=len,ans=-1;
        while(l!=r){
            int mid=(l+r-1)>>1;
            if(calc(b[mid])>=m) ans=b[mid],l=mid+1;
            else r=mid;
        }
        printf("%d\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40679299/article/details/81274486