This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
#define break_it assert(false)
int main(void) {
ios::sync_with_stdio(0);
cin.tie(0);
int n, k;
cin >> n >> k;
vector <int> a(n+k);
for(int i = 0; i < n +k; i ++) cin >> a[i];
function <int(vector <int>)> check=[](vector <int> s) {
int n = s.size();
for(int i = 0; i < n; i ++) {
int j = i + 1;
if(j >= n or n - i - 1 < 0 or n - j - 1 < 0) break;
if(s[i] + s[n-i-1] == s[j]+s[n-j-1]) continue;
return 0;
}
return 1;
};
function <void(int, int)> solve=[&](int start, int end) {
vector <int> fors;
int sum = a[start] + a[end];
int none = -1;
int l = 0, r = n+k-1;
while(l < r){
if(a[l] + a[r] > sum) {
if(none != -1) return;
none = r;
r--;
}else if(a[l] + a[r] < sum) {
if(none != -1) return;
none = l;
l ++;
}
l ++;
r --;
}
if(none == -1) {
none = (n+k)/2;
}
for(int i = 0; i <= n+k-1; i ++) {
if(none == i) continue;
fors.push_back(a[i]);
}
for(auto i : fors) cout << i << " ";
exit(0);
};
function <void(int, int)> solve2=[&](int start, int end) {
vector <int> fors;
int sum = a[start] + a[end];
int none = -1, none2 = -1;
int l = 0, r = n+k-1;
while(l < r){
if(a[l] + a[r] > sum) {
if(none != -1 and none2 != -1) return;
if(none == -1) {
none = r;
}else none2 = r;
r--;
}else if(a[l] + a[r] < sum) {
if(none != -1 and none2 != -1) return;
if(none == -1) {
none = l;
}else none2 = l;
l ++;
}
l ++;
r --;
}
for(int i = 0; i <= n+k-1; i ++) {
if(none == i or i == none2) continue;
fors.push_back(a[i]);
}
for(auto i : fors) cout << i << " ";
exit(0);
};
/*if(n+k<=18) {
for(int mask = 0; mask < (1 << (n+k)); mask ++) {
if(__builtin_popcount(mask) == n) {
vector <int> v;
for(int i = 0; i < n+k; i ++) {
if((mask & (1 << i))!=0) {
v.push_back(a[i]);
}
}
if(check(v)) {
for(int i = 0; i < v.size(); i ++) cout << v[i] << " ";
return false;
}
}
}
return false;
}else if(k == 1){
solve(0, n+k-1);
solve(0, n+k-2);
solve(1, n+k-1);
break_it;
}else */if(k == 2) {
for(int i = 0; i < n+k-4; i ++) {
solve2(i, i+4);
solve2(i, i+3);
solve2(i, i+2);
solve2(i, i+1);
}
break_it;
}
return false;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |