K-th Number HDU - 6231

http://acm.hdu.edu.cn/showproblem.php?pid=6231

看到题目就想数据结构 想算贡献 智障

二分答案 把大于等于当前二分值的数对应位置都置为1 其余置为0

然后枚举i 如果i和j确定的这个区间里不够k个数 就把j往后挪 然后[i,j] [i,j+1] [i,j+2]...[i,n]都是符合条件的区间 累加一下 看最后满足条件的区间够不够m

 

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;

ll sum[maxn];
int ary[maxn],tmp[maxn];
ll m;
int n,k,len;

bool judge(int lim)
{
    ll res;
    int i,j;
    for(i=1;i<=n;i++){
        sum[i]=sum[i-1];
        if(ary[i]>=lim) sum[i]++;
    }
    res=0,j=1;
    for(i=1;i<=n;i++){
        while(j<=n&&sum[j]-sum[i-1]<k) j++;
        res+=(n-j+1);
    }
    if(res>=m) return 1;
    else return 0;
}

int main()
{
    int t,i,l,r,mid,ans;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%lld",&n,&k,&m);
        for(i=1;i<=n;i++){
            scanf("%d",&ary[i]);
            tmp[i]=ary[i];
        }
        sort(tmp+1,tmp+n+1);
        len=unique(tmp+1,tmp+n+1)-tmp-1;
        l=1,r=len;
        while(l<=r){
            mid=(l+r)/2;
            if(judge(tmp[mid])){
                l=mid+1,ans=tmp[mid];
            }
            else{
                r=mid-1;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}