NC14301 K-th Number

题目链接

题目链接

题意

给定长度为N的数组A，对所有长度大于等于k的连续子段，取出其第k大放入数组 B中。求数组B的第M大。

思路

• 利用二分求解答案，对端点中点进行判断，判断第k个数比中点数大的区间个数
• 记得开long long！！！！

参考代码

#include<bits/stdc++.h>
using namespace std;
long long aa[100010];
long long n,m,k;
long long judge(long long s)
{
    long long re=0,num=0;
    int j=0;
    for(int i=0; i<n; i++)
    {
        if(aa[i]>=s)
            num++;
        if(num==k)
        {
            re+=n-i;
            while(aa[j]<s&&j<n)
            {
                re+=n-i;
                j++;
            }
            num--;
            j++;
        }
    }
    return re>=m;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n>>k>>m;
        long long l=1,r=1e9+10,ans=0;
        for(int i=0; i<n; i++)
            cin>>aa[i];
        while(l<=r)
        {
            long long mid=(l+r)/2;
            if(judge(mid))
            {
                ans=mid;
                l=mid+1;
            }
            else
            {
                r=mid-1;
            }
        }
        cout<<ans<<endl;
    }
}