gym101234G(优先队列)

7 篇文章 0 订阅
3 篇文章 0 订阅

题意:给一堆数字,求第K大的组合(每一位数字不重复用)

最开始用的multiset,超时。提醒后想了优先队列,可惜一个数只能加一次这个问题并没有得到解决,后来看了大佬的代码,发现自己少想了一步,就是在优先队列当中加入这个数字本来在数组当中的位置。这样,对于每个当前数字,便有选与不选两种情况,first代表组合大小,second代表当前处理到了第几个数字。用second来记录选到了哪个数字。

#include<bits/stdc++.h>
//#define 3e6+7 MAXN
using namespace std;
typedef pair<long long,int> PLLI;
priority_queue<PLLI,vector<PLLI>,greater<PLLI> > s;
long long ts[200007];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    int k,n;
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>ts[i];
    sort(ts+1,ts+n+1);
    s.push(PLLI(ts[1],1));
    int tot=0;
    while(!s.empty()){
        PLLI temp=s.top();
        s.pop();
        long long a=temp.first;
        tot++;
        if(tot==k){
            cout<<a<<endl;
            return 0;
        }
        if(temp.second+1<=n){ //这个条件不能省略,一旦省略后当k略大于n时会push入0
            s.push(PLLI(a+ts[temp.second+1],temp.second+1));
            s.push(PLLI(a+ts[temp.second+1]-ts[temp.second],temp.second+1));
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值