# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
947807 | biank | A Plus B (IOI23_aplusb) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "aplusb.h"
using namespace std;
#define sz(x) int(x.size())
using ii = pair<int, int>;
using vi = vector<int>;
vi smallest_sums(int N, vi A, vi B) {
priority_queue<pair<int, ii>, vector<pair<int, ii>>, greater<pair<int, ii>>> pq;
set<ii> visited;
auto push = [&](int i, int j) {
pq.emplace(a[i] + b[j], {i, j});
vis.emplace(i, j);
};
push(0, 0);
vi ans;
while (sz(ans) < N) {
auto [sum, pos] = pq.top();
ans.push_back(sum);
pq.pop();
auto [i, j] = pos;
if (i + 1 < N && !vis.count({i + 1, j})) {
push(i + 1, j);
}
if (j + 1 < N && !vis.count({i, j + 1})) {
push(i, j + 1);
}
}
return {};
}