题意:给一堆数字,求第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;
}