其他分享
首页 > 其他分享> > HDU - 6231 K-th Number

HDU - 6231 K-th Number

作者:互联网

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=6231

题意:给出一个数组a 将其所有子段中的第k大的数字放进b中,求b中的第m大的数字。

思路:二分答案,尺取check,尺取的目的是判断大于等于x为第k大的区间有多少个。枚举每一个左端点l,使得整个区间内的大于等于x的数为k个,此时符合条件的区间数要加上(n-r) 
因为当右端点在[r,n-1]这个区间内时,去k大值一定是大于等于x的。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath> 
#include<map>
#include<queue>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
int N,K;
ll M;
int a[100005];
bool check(int x)
{
    ll ans=0;
    int num=0;
    int j=1; 
    for(int i=1;i<=N;i++)
    {
        if(a[i]>=x)
            num++;
        if(num==K)
        {
            ans+=N-i+1;//统计后面一共可以形成多少个区间
            while(a[j]<x)
            {
                ans+=N-i+1;//统计前面一共可以形成多少个区间
                j++;
            }
            num--;//还原状态
            j++;
        }
    }
    if(ans>=M)
        return true;
    else
        return false; 
} 
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]);
        int l=1,r=1000000000;
        int m;
        while(l<r)
        {
            m=(l+r)/2;
            if(check(m))
                l=m+1;
            else
                r=m;
        }
        printf("%d\n",l-1);
    }
    return 0;
}

 

标签:HDU,int,ll,Number,num,区间,尺取,include,6231
来源: https://www.cnblogs.com/2462478392Lee/p/12426164.html