Submission #524506

#	Time	Username	Problem	Language	Result	Execution time	Memory
524506		maks007	Table Tennis (info1cup20_tabletennis)	C++14	20 / 100	3081 ms	4572 KiB

tabletennis

#include <bits/stdc++.h>
 
using namespace std;
 #define break_it (1/0)
int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	int n, k;
	cin >> n >> k;
	vector <int> a(n+k);
	function <int(vector <int>)> check=[&](vector <int> s) {
		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;
	};
	for(int i = 0; i < n+k; i ++) cin >> a[i];
	if(k == 1) { 
		for(int i = 0; i < n + k; i ++) {
			vector <int> v;
			for(int j = 0; j < n + k; j ++) {
				if(i == j) continue;
				v.push_back(a[j]);
			}
			if(check(v)){
				for(auto i : v) cout << i << " ";
				return false;
			}
		}
	}else 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;
				}
			}
		}
	}else if(k == 1){
		for(int i = 0; i <= 1; i ++) {
			vector <int> v;
			for(int j = 0; j < n + k; j ++) {
				if(i == j) continue;
				v.push_back(a[j]);
			}
			if(check(v)){
				for(auto i : v) cout << i << " ";
				return false;
			}
		}
		for(int i = n - 1; i >= n -2; i --) {
			vector <int> v;
			for(int j = 0; j < n + k; j ++) {
				if(i == j) continue;
				v.push_back(a[j]);
			}
			if(check(v)){
				for(auto i : v) cout << i << " ";
				return false;
			}
		}
	}
	return false;
}  

Compilation message (stderr)

tabletennis.cpp: In function 'int main()':
tabletennis.cpp:43:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |      for(int i = 0; i < v.size(); i ++) cout << v[i] << " ";
      |                     ~~^~~~~~~~~~

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