K-th Number (二分+尺取) HDU6231

最新推荐文章于 2025-03-02 22:34:04 发布

mizp

最新推荐文章于 2025-03-02 22:34:04 发布

阅读量334

点赞数

分类专栏：二分尺取

本文链接：https://blog.youkuaiyun.com/qq_42819598/article/details/100918379

版权

二分尺取专栏收录该内容

2 篇文章

订阅专栏

给定数组A，需要找到所有长度不小于K的子区间内的第K大元素，这些元素组成新的数组B。博客介绍了如何反向利用二分查找策略，高效地求解数组B的第M大元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

K-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 find this number.
N<=1e5
题意: 给你A数组，找出A数组中的所有子区间的第K大，所有第K大组成B数组，输出B数组第M大
思路: 我们要正着找出第M大复杂度N^2,但是我们可以反着来假设第M大，不断通过二分check正确性

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
int n,k;
ll m;
int a[N];
bool check(ll x)
{
    ll cnt=0,ans=0;
    int l=1;
    for(int i=1;i<=n;i++)
    {
        if(a[i]>=x)
            cnt++;
        if(cnt==k)
        {
            ans+=n-i+1;
            while(a[l]<x&&l<=i)
            {
                ans+=n-i+1;
                l++;
            }
            l++;
            cnt--;
        }
    }
    if(ans>=m)
        return 1;
    return 0;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%lld",&n,&k,&m);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        ll l=1,r=1e10;
        while(l<r)
        {
            ll mid=(l+r)>>1;
            if(check(mid))
            {
                l=mid+1;
            }
            else
                r=mid;
        }
        printf("%lld\n",l-1);
    }
    return 0;
}