Kth Takoyaki Set (AtCoder abc297_e)

原题链接E - Kth Takoyaki Set (atcoder.jp)https://atcoder.jp/contests/abc297/tasks/abc297_e

思路：

根据题目数据范围

• 1≤K≤2×
• 那抹我们考虑从小到大枚举即可
• 为了保证每次的数尽量小，我们使用优先队列进行优化
• 用bfs即可

代码 ：
 

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;
typedef long long ll;
int n,k,cnt;
ll a[15];
ll s[5000005];
unordered_map<ll,bool>mp;
priority_queue<ll,vector<ll>,greater<ll> >q;
void bfs(){
	for(int i=1;i<=n;i++){
		if(mp[a[i]]==0){
			q.push(a[i]);
			mp[a[i]]=1;
			cnt++;
			s[cnt]=a[i];
		}
	}
	while(!q.empty()){//
		ll x=q.top();
		q.pop();
		for(int i=1;i<=n;i++){
			if(mp[x+a[i]]==0){
				q.push(x+a[i]);
				mp[x+a[i]]=1;
				cnt++;
				s[cnt]=x+a[i];
			}
		}
		if(cnt>k*9) break;
	}
}
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++) cin>>a[i];
    sort(a+1,a+1+n);
	bfs(); 
	int c1=0;
	sort(s+1,s+1+cnt);
	cout<<s[k];
}