题意:给你一个n个数的数组,把这个数组中长度大于等于k的区间中第k大的数取出来组成一个新数组,问你这个新数组第m大是多少。
可能有的人和我一样没去分开第k大和第k小的关系,比如数组[1,2,3,4,5],第2大就是4,第2小就是2.
思路:我们可以用尺取算出第k大大于等于x的区间个数有多少个。当我们了解这个之后自然能想到二分答案。
我们枚举区间右端点r,然后判断当前大于等于x的个数如果大于等于k个,那么l不变,r向右的区间的第k大都大于等于x。当前计数加上ans+=n-r+1即可。如果ans>=m那么说明区间第k大大于等于x的区间个数大于等于m个,这说明我们的mid比实际答案要小。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[100005];
ll n,k,m;
ll check(ll x)
{
ll ans=0,num=0;
for(ll l=1,r=1;r<=n;r++)
{
if(num<k)
{
if(a[r]>=x) num++;
}
while(num>=k&&l<=r)
{
ans+=n-r+1;
if(a[l]>=x)
{
num--;
}
l++;
}
}
return ans>=m;
}
int main()
{
ll t;scanf("%lld",&t);
while(t--)
{
scanf("%lld%lld%lld",&n,&k,&m);
for(ll i=1;i<=n;i++) scanf("%lld",&a[i]);
//cout<<check(1)<<endl;
ll l=1,r=1e9+7;
while(l<=r)
{
ll mid=(l+r)/2;
if(check(mid)) //如果mid第k大的区间个数大于m个,说明mid比实际答案小。
{
l=mid+1;
}
else
{
r=mid-1;
}
}
printf("%lld\n",r);
}
return 0;
}