Submission #423605

#	Time	Username	Problem	Language	Result	Execution time	Memory
423605		pavement	Table Tennis (info1cup20_tabletennis)	C++17	72 / 100	3123 ms	371632 KiB

tabletennis

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
#ifdef _WIN32
#define getchar_unlocked _getchar_nolock
#endif
#define int long long
#define mp make_pair
#define mt make_tuple
#define pb push_back
#define ppb pop_back
#define eb emplace_back
#define g0(a) get<0>(a)
#define g1(a) get<1>(a)
#define g2(a) get<2>(a)
#define g3(a) get<3>(a)
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
typedef double db;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> ii;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef tree<iii, null_type, greater<iii>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

int N, K, A[200005];
map<int, int> R;

main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cin >> N >> K;
	for (int i = 1; i <= N + K; i++) cin >> A[i], R[A[i]] = i;
	for (int i = 0; i <= K; i++)
		for (int j = 0; i + j <= K; j++) {
			vector<int> ans;
			int S = A[i + 1] + A[N + K - j];
			for (int k = i + 1; k <= N + K - j; k++) {
				if (S - A[k] > A[k] && R[S - A[k]] && R[S - A[k]] <= N + K - j) {
					ans.pb(k);
					ans.pb(R[S - A[k]]);
				}
			}
			if (ans.size() >= N) {
				vector<int> tmp;
				for (int i = 0; i < N; i++) tmp.pb(A[ans[i]]);
				sort(tmp.begin(), tmp.end());
				for (int i : tmp) cout << i << ' ';
				cout << '\n';
				return 0;
			}
		}
	//assert(0);
}

Compilation message (stderr)

tabletennis.cpp:31:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   31 | main() {
      | ^~~~
tabletennis.cpp: In function 'int main()':
tabletennis.cpp:46:19: warning: comparison of integer expressions of different signedness: 'std::vector<long long int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   46 |    if (ans.size() >= N) {
      |        ~~~~~~~~~~~^~~~

• 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: 0 / 15
• Subtask 8: 0 / 13