Kth Takoyaki Set (AtCoder abc297_e)

原题链接E - Kth Takoyaki Set (atcoder.jp)icon-default.png?t=N2N8https://atcoder.jp/contests/abc297/tasks/abc297_e

思路:

根据题目数据范围

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

代码 :
 

#include
#include
using namespace std;
typedef long long ll;
int n,k,cnt;
ll a[15];
ll s[5000005];
unordered_mapmp;
priority_queue,greater >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<

你可能感兴趣的:(c++,算法)