刷题记录:牛客NC14301K-th Number

传送门:牛客

题目描述:

Alice are given an array A[1..N] with N numbers.
Now Alice want to build an array B by a parameter K as following rules:
Initially, the array B is empty. Consider each interval in array A. If the length of this interval is less than K, then 
ignore this interval. Otherwise, find the K-th largest number in this interval and add this number into array B.
In fact Alice doesn't care each element in the array B. She only wants to know the M-th largest element in the 
array B. Please help her to fi nd this number.
输入:
2
5 3 2
2 3 1 5 4
3 3 1
5 8 2
输出:
3
2

这道题的题面有一点绕,思路也比较难想吧

主要思路:

1. 刚开始我试了试贪心的做法,然后我发现有一个这样的事实,就是当我们有符合的数的区间之后,在该区间的后面增加几个数字其实对于该区间的影响永远都是正影响,原因是因为当你增加了一个数A,假设A的大小比之前排在k位的大,那么现在的较大值变成了k,比原来优,假设A的值比原来的小,那么现在的较大值还是K,并没有产生负影响,所以当时我一想到这个优美的性质,我焯直接开干
2. 我们现在知道了思路一中的优美性质,我们就会发现显然长度越长的区间的第K大的值显然是越大的,并且区间长度为N的有一种,区间长度为N-1有2中,并且依次递增,然后就没有然后了,你以为我们是不是就可以开干了,其实并不是,因为我们并不能将其排序,所以我们的贪心思想是错的,但是不要慌,这个优美性质对于问题还是有帮助的
3. 接下来就是我们的二分+尺取法思路了(其实比较难想),当时我在模拟这道题时是想到了尺取法的,但是缺少了二分导致我一直卡住(算了,话不多说,下面开始介绍这种方法)
4. 我们先二分出最终排在M位的元素(当然此时我们是假定的,并不一定是最终答案),然后根据我们二分答案的一般性套路,接下来就考虑check函数即可.对于一个区间,我们先找到包含了K个大于我们的mid值的区间(显然此时我们的区间是合法的,并且有),对于这个区间我们使用之前的性质,显然当我们的右区间不断右移时,我们得到的每一个区间显然都是符要求的,并且我们有N-R+1组这样的区间,加上本身一共N-R组,然后我们再将左区间一步步移动到离他最近的大于mid的位置即可(当然每移一步都要加上右边的可扩展部分)

注意点:要使用longlong,注意二分模板的写法

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
ll n,k,m,a[maxn];
ll check(ll mid) {
	ll cnt=0;ll ans=0;
	ll l=1,r=1;
	while(l<=n){
		if(a[r]>=mid){
			cnt++;
			if(cnt==k) {
				ans+=(n-r+1);
				while(a[l]<mid) {
					l++;
					ans+=(n-r+1);	
				}
				l++;
				cnt--;
			} 
		}
		r++;
		if(r==n+1) break;
		
	}
	if(ans>=m) return true;
	else return false;
}
int main() {
	ll T;T=read();
	while(T--) {
		n=read();k=read();m=read();
		memset(a,0,sizeof(a));
		ll l=1,r=-1;
		for(int i=1;i<=n;i++){
			a[i]=read();
			r=max(r,a[i]);
		}
		ll ans;
		while(l<=r){
			ll mid=(l+r)>>1;
			if(check(mid)) {
				ans=mid;
				l=mid+1;
			}else {
				r=mid-1;
			}
		}
		printf("%lld\n",ans);
	}
	return 0;
}