Submission #499882

#	Time	Username	Problem	Language	Result	Execution time	Memory
499882		jesus_coconut	Table Tennis (info1cup20_tabletennis)	C++17	87 / 100	3078 ms	5152 KiB

tabletennis

#include <bits/stdc++.h>
#define pb push_back
#define all(a) begin(a), end(a)
using namespace std;

int n, k;
vector<int> v;
vector<int> dif;
void load() {
	cin >> n >> k;
	v.resize(n + k);
	for (auto &it : v) cin >> it;
}

vector<int> ans;

bool solve(int l, int r) {
	if (l >= r) return false;
	ans.clear();
	ans.pb(v[l]);
	ans.pb(v[r]);
	while (l < r) {
		int cl = dif[l+1];
		int cr = dif[r];
		int pl = l, pr = r;
		while (cl != cr && l < r) {
			if (cl < cr) {
				l++;
				cl += dif[l+1];
			} else {
				r--;
				cr += dif[r];
			}
		}
		++l; --r;
		if (l >= r) break;
		if (cl == cr) {
			ans.pb(v[l]);
			ans.pb(v[r]);
		}
		if (ans.size() == n) return true;
	}
	return false;
}

void solve() {
	for (int i = 0; i <= k; ++i) {
		for (int j = 0; j <= k - i; ++j) {
			if (solve(i, n + k - 1 - j)) {
				sort(all(ans));
				for (auto &it: ans) cout << it << ' ';
				return;
			};
		}
	}
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

	load();
	sort(all(v));
	dif.resize(size(v));
	adjacent_difference(all(v), begin(dif));
	solve();

	return 0;
}

Compilation message (stderr)

tabletennis.cpp: In function 'bool solve(int, int)':
tabletennis.cpp:41:18: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   41 |   if (ans.size() == n) return true;
      |       ~~~~~~~~~~~^~~~
tabletennis.cpp:25:7: warning: unused variable 'pl' [-Wunused-variable]
   25 |   int pl = l, pr = r;
      |       ^~
tabletennis.cpp:25:15: warning: unused variable 'pr' [-Wunused-variable]
   25 |   int pl = l, pr = r;
      |               ^~

• Subtask 1: 11 / 11
• Subtask 2: 9 / 9
• Subtask 3: 14 / 14
• Subtask 4: 15 / 15
• Subtask 5: 9 / 9
• Subtask 6: 14 / 14
• Subtask 7: 15 / 15
• Subtask 8: 0 / 13