Submission #900643

# Submission time Handle Problem Language Result Execution time Memory
900643 2024-01-08T19:01:56 Z mannshah1211 A Plus B (IOI23_aplusb) C++17
0 / 100
1000 ms 432 KB
#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 time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 432 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Execution timed out 1026 ms 344 KB Time limit exceeded
5 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 432 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Execution timed out 1026 ms 344 KB Time limit exceeded
5 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 432 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Execution timed out 1026 ms 344 KB Time limit exceeded
5 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 432 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Execution timed out 1026 ms 344 KB Time limit exceeded
5 Halted 0 ms 0 KB -