hdu 6231 (二分+尺取)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_37632935/article/details/84776913

题意:给你一个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;
}

猜你喜欢

转载自blog.csdn.net/qq_37632935/article/details/84776913