K-th Number题解

K-th Number

https://ac.nowcoder.com/acm/problem/14301

题意:就是给你N个树,然后你去把这些数分成很多个区间然后把这些区间的找到他的第K大,然后这些第K大的数又组合成一个新的数组,最后找出这个新的数组里面的第M大。
苦苦想了很久都不知道如何做,还是看了题解才豁然开朗。(太菜了)
其实就是二分+尺取法,首先就是二分一个答案,然后每次去找第K大,我们可以开一个cnt记录当前这个答案在原数组的尺取区间是第K大,然后cnt==k 我们就应该找去可以满足的区间数,比如当前一个满足区间是1-i,那么是不是i-n这里面可以一直延续到最后一个数,每次形成一个新的区间,那么是不是就有 n-i+1个这样的区间,自己可以好好想一下,当然这是大范围,列如i=10,那么1-10这个区间就已经有第K大的数了,但是有可能1-4的区间的数组小于了我们的答案,是不是我们就可以2-i,3-i,4-i,这些又是一个新区间,但是答案却不变,依旧是n-i+1,所以每次我们还要在cnt=k的时候while一下左边的端点是不是有小于第k大的数,最后我们所有的区间数和M比较一下,如果小于M,那么说明在新数组里面找不到第M大,反正就有,而且你的二分答案,将会是最优。
PS:最后为啥输出l-1呢?

da:如果当前的mid符合情况,mid的有可能是最终的答案,但是我写的那里让l = mid + 1了,所以最后一次,得到的mid为最终答案的时候,l又等于mid+1了,所以最后mid为答案,l=mid+1,所以输出l-1

#include <bits/stdc++.h>
#define fio ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define debug(x) cout << #x << ": " << x << endl;
#define debug1(x) cout<<"xxx"<<endl;
#define ll long long
#define ull unsigned long long
#pragma GCC optimize("Ofast","inline","-ffast-math")
#pragma GCC target("avx,sse2,sse3,sse4,mmx")
#define mse(a,b) memset(a,b,sizeof b);
using namespace std;
const int maxx=1e6+100;
const int mod=1e9+7;
ll ans[maxx];
ll n,m,k;
ll judge(ll x)
{
    ll sum=0,cnt=0,j=1;
    for(int i=1;i<=n;i++)
    {
        if(ans[i]>=x) cnt++;
        if(cnt==m)
        {
            sum+=n-i+1;
            while(ans[j]<x)
            {
                j++;
                sum+=n-i+1;
            }
            j++;
            cnt--;
        }
    }
    if(sum>=k)
        return 1;
    else
        return 0;
}
int main()
{
    int a;
    cin>>a;
    while(a--){
    cin>>n>>m>>k;
    ll ma=-1e9;
    for(int i=1;i<=n;i++)
    cin>>ans[i];//ma=max(ma,ans[i]);
    ll l=1,r=2e9;
    while(l<r)
    {
        ll mid = l + r >> 1;
        if(judge(mid)) l=mid+1;
        else r=mid;
    }
    cout<<l-1<<endl;
    }
    return 0;
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务