# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1123340 | njoop | A Plus B (IOI23_aplusb) | C++17 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
#define ll long long
using namespace std;
vector<int> ans;
vector<int> smallest_sums(int N, vector<int> A, vector<int> B) {
pq.push({A[0]+B[0], 0, 0});
ll l=0, r=2e9+1, val=r;
while(l < r) {
ll mid = l+(r-l)/2, cnt = 0;
for(int i=0; i<N; i++) {
cnt += upper_bound(B.begin(), B.end(), mid-A[i])-B.begin();
}
if(cnt >= N) {
r = mid-1;
val = mid;
} else {
l = mid+1;
}
}
for(int i=0; i<N; i++) {
for(int j=0; A[i]+B[j] <= val; j++) {
ans.push_back(A[i]+B[j]);
}
}
sort(ans.begin(), ans.end());
while(ans.size() > N) ans.pop_back();
return ans;
}