Submission #840490

#	Time	Username	Problem	Language	Result	Execution time	Memory
840490		ZeroCool	A Plus B (IOI23_aplusb)	C++17	100 / 100	95 ms	12792 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) {
	priority_queue<pair<int, pair<int, int>>, vector<pair<int, pair<int, int>>>, greater<>> pq;
	vector<int> ans;
	set<pair<int, int>> vis;
	pq.push({A[0] + B[0], {0, 0}});
	for (int i = 0;i<n;i++){
		auto h = pq.top();
		pq.pop();
		ans.push_back(h.first);
		if (!vis.count({h.second.first + 1, h.second.second})){
			pq.push({A[h.second.first + 1] + B[h.second.second], {h.second.first + 1, h.second.second}});
			vis.insert({h.second.first + 1, h.second.second});
		}
		if (!vis.count({h.second.first, h.second.second + 1})){
			pq.push({A[h.second.first] + B[h.second.second + 1], {h.second.first, h.second.second + 1}});
			vis.insert({h.second.first, h.second.second + 1});
		}
	}
	return ans;
}

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