제출 #900643

#제출 시각아이디문제언어결과실행 시간메모리
900643mannshah1211A Plus B (IOI23_aplusb)C++17
0 / 100
1026 ms432 KiB
#include "aplusb.h"
#include <bits/stdc++.h>

std::vector<int> smallest_sums(int n, std::vector<int> a, std::vector<int> b) {
	std::function<int(int)> count_smaller = [&](int sum) {
		int cnt = 0;
		for (int i = 0; i < n; i++) {
			int low = 0, high = n - 1, index = -1;
			while (low <= high) {
				int middle = (low + high) / 2;
				if (a[i] + b[middle] <= sum) {
					index = middle;
					low = middle + 1;
				} else {
					high = middle - 1;
				}
			}
			cnt += (index + 1);
			if (cnt > n) {
				return cnt;
			}
		}
		return cnt;
	};
	int low = 0, high = 2E9, index = -1;
	while (low <= high) {
		int middle = (low + high) / 2;
		if (count_smaller(middle) >= n) {
			index = middle;
			high = middle - 1;
		} else {
			low = middle + 1;
		}
	}
	std::vector<int> answer;
	int taken = 0;
	for (int i = 0; i < n; i++) {
		int low = 0, high = n - 1, ind = -1;
		while (low <= high) {
			int middle = (low + high) / 2;
			if (a[i] + b[middle] < index) {
				ind = middle;
				low = middle + 1;
			} else {
				high = middle - 1;
			}
		}
		for (int j = 0; j <= ind; j++) {
			answer.push_back(a[i] + b[j]);
		}
		taken += (ind + 1);
	}
	for (int j = 0; j < n - taken; j++) {
		answer.push_back(index);
	}
	std::sort(answer.begin(), answer.end());
	return answer;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...