K-th Number（牛客）二分

最新推荐文章于 2022-01-04 18:23:56 发布

原创最新推荐文章于 2022-01-04 18:23:56 发布 · 188 阅读

0 ·

CC 4.0 BY-SA版权

该博客介绍了如何运用二分搜索算法解决牛客网上的一道题目，涉及数组区间和第k大数的概念。题目要求找出数组a中所有长度大于等于k的区间的第k大数构成的数组b中的第m大数。博主通过二分答案，判断包含至少k个不小于mid的区间的数量来求解。

传送门：https://ac.nowcoder.com/acm/problem/14301
打死我也想不到是二分。

题目大意：
数组b的元素是，数组a中所有区间长度大于等于k的区间的第k大数(有点绕？)
求数组b中第m大的数。

二分答案，judje的内容是求所有区间中包含大于等于k个不小于mid的区间的数量

//#pragma GCC optimize(3,"Ofast","inline")
#include<unordered_map>
#include<unordered_set>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<functional>
#include<cstring>
#include<string>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
#include<vector>
#include<sstream>
#include<list>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int maxn= 2e7+10;
const ll mod=6662333;
const int N=1e5+10;
int n,k;
ll m;
int a[N];
int s[N];
int judje(int x)
{
    ll l=0,sum=0;
    for(int i=1;i<=n;i++)s[i]=s[i-1]+(a[i]>=x);
    for(int i=k;i<=n;i++)
    {
        //这个循环里的求法真是绝了，本来o（n2—）解决的问题直接o（n）解决
        //我好菜~
        while(s[i]-s[l]>=k)l++;
        sum+=l;
    }
    return sum>=m;
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        cin>>n>>k>>m;
        int l=1,r=0;
        for(int i=1;i<=n;i++){cin>>a[i];r=max(r,a[i]);}
        while(l<r)
        {
            int mid=(l+r+1)>>1;
            if(judje(mid))l=mid;
            else r=mid-1;
        }
        cout<<l<<endl;
    }
    return 0;
}