제출 #1357824

#제출 시각아이디문제언어결과실행 시간메모리
1357824whallyA Plus B (IOI23_aplusb)C++20
60 / 100
1129 ms1147816 KiB
#include <bits/stdc++.h>
#include "aplusb.h"
using namespace std;
#define ll long long
int oa[100010], ob[100010];

vector<int> smallest_sums(int n, vector<int> a, vector<int> b)
{

	vector<int> ans;
	a.insert(a.begin(), 0);
	b.insert(b.begin(), 0);
	for (int i = 1; i <= n; i++){
		oa[i] = a[i];
		ob[i] = b[i];
	}
	int l = 1, r = n;
	int res = -1;
	while (l <= r){
		int mid = (l+r)/2;
		if (mid < n && (n+1)-mid < n){
			if (a[mid+1] < b[(n+1)-mid]) l = mid+1;
			else if (b[(n+1)-mid+1] < a[mid]) r = mid-1;
			else {
				res = mid;
				break;
			}
		}
		else if (mid == n){
			if (b[(n+1)-mid+1] < a[mid]) r = mid-1;
			else {
				res = mid;
				break;
			}
		}
		else if ((n+1)-mid == n){
			if (a[mid+1] < b[(n+1)-mid]) l = mid+1;
			else {
				res = mid;
				break;
			}
		}
	}
	if (res == -1) res = l;
	vector<int> all;
	for (int i = 1; i <= res; i++){
		for (int j = 1; j <= (n+1)-res; j++){
			all.push_back({oa[i] + ob[j]});
		}
	}
	sort(all.begin(), all.end());
	for (int i = 0; i < n; i++){
		ans.push_back(all[i]);
	}
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...