题意:所有区间长度>=k的连续区间中第k大的数字组成一个集合,求最终集合中第m大的数字。
思路:一度以为是主席树,但主席树又不会用,,想了半天也没思路。看了题解,感觉这个解法真的很巧妙,自己想的话是想不到二分的。主要思想是二分答案,即最终集合中第m大的那个值,若当前二分的值为x,求出满足区间第k大元素的值>=x的区间有多少个,若比m多,说明我们的x小了,保留右半区间,否则保留左半区间。
至于求满足条件的区间有多少个的过程,就是尺取,区间合法必须满足的条件是,区间内>=x的元素个数num>=k,那么如果l,r区间满足了这个条件,[l, r+1], [l, r+2] … 这些区间都满足,那么直接答案+上n-r+1,再把l往右移即可,但如果这一步移动导致我们少了一个>=x的元素,就要把r接着右移,直到满足条件为止。
对了,还有一个大坑点!吐血,,题目没有告诉m的范围!!我就按int读的,wa到怀疑人生,fo了
代码:
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
typedef long long ll;
const int N = 1e5+10,inf=1e9;
int a[N];
int n,k;
ll m;
bool check(int x)
{
int l=1,r=0,num=0;
ll ans=0;
while(r<=n)
{
while(num<k&&r<=n)
{
if(a[++r]>=x) num++;
}
if(num<k) break;
ans+=n-r+1;
if(a[l]>=x) num--;
l++;
}
return ans>=m;
}
int main()
{
int t;
cin>>t;
while(t--)
{
scanf("%d%d%lld",&n,&k,&m);
int l=1,r=inf,mid;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
while(l<r)
{
mid=l+r+1>>1;
if(check(mid))
l=mid;
else
r=mid-1;
}
printf("%d\n",l);
}
return 0;
}