Submission #840486

#	Time	Username	Problem	Language	Result	Execution time	Memory
840486		ZeroCool	A Plus B (IOI23_aplusb)	C++17	10 / 100	1 ms	212 KiB

aplusb

#include "aplusb.h"
#include <bits/stdc++.h>

using namespace std;

#define ll long long

vector<int> smallest_sums(int n, vector<int> A, vector<int> B) {
	set<int >s;
	priority_queue<tuple<int,int,int> , vector<tuple<int,int,int> > , greater<tuple<int,int,int> > > pq;


	pq.push({A[0] + B[0], 0, 0});
	set<pair<int,int> > st;
	vector<int> ans;
	for(int i = 0;i<n;i++){
		int val = get<0>(pq.top());
		int posa = get<1>(pq.top());
		int posb = get<1>(pq.top());
		pq.pop();

		ans.push_back(val);

		if(posa + 1 < n && !st.count({posa + 1, posb})){
			pq.push({A[posa + 1] + B[posb],posa+1, posb});
			st.insert({posa+1, posb});
		}

		if(posb + 1 < n && !st.count({posa, posb+1})){
			pq.push({A[posa] + B[posb+1],posa, posb+1});
			st.insert({posa, posb+1});
		}
	}
	
	return ans;
}

• Subtask 1: 10 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 30
• Subtask 4: 0 / 40