Question

将所有连续区间中第K大的数放入一个新的数组B中,求数组B中第M大的数为多少?

Solution

二分+尺取
答案具有单调性,故考虑二分。看了邓老师的题解之后发现居然可以二分诶
那么问题就是check函数该如何写了,这里要用到尺取法判断。
我们求任意连续区间中第K大的数的数量,若数量则L=mid+1,反之R=mid。
当一个连续区间中有K个数的时候,他的右边界剩余数的数量与其组成的连续区间均满足,故ans+=n-r+1记录。
然后左指针更新。

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const int INF  = 0x3f3f3f3f;
const ll  mod  = 1e9 + 7;
const ll  maxn = 1e6 + 5;
const int N = 1e5 + 5;

ll n,k,m,a[N];

bool check(ll x){
    ll l=1,r=0,num=0;
    ll ans=0;
    while(l<=n){
        while(r<n && num<k)
            if(a[++r]>=x) num++;
        if(num==k) ans+=n-r+1;
        if(a[l]>=x) num--;
        ++l;
    }
    return ans>=m;
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;cin>>T;
    while(T--){
        cin>>n>>k>>m;
        for(int i=1;i<=n;i++) cin>>a[i];
        ll L=1,R=1e9+1,mid,ans=0;
        while(L<R){
            mid=(L+R)/2;
            if(check(mid)) L=mid+1,ans=mid;
            else R=mid;
        }
        cout<<ans<<'\n';
    }
    return 0;
}