oj.uz

답안 #365735

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
365735 2021-02-12T09:44:48 Z tengiz05 Table Tennis (info1cup20_tabletennis) C++17
44 / 100
 3000 ms 622304 KB 
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define FASTIO ios_base::sync_with_stdio(false); cin.tie(NULL);
#define all(x) (x).begin(), (x).end()
#define pb push_back
#define pii pair<int, int>
#define ff first
#define ss second
#define PI acos(-1)
#define ld long double
template<class T> bool chmin(T& a, const T& b) {return a>b? a=b, true:false;}
template<class T> bool chmax(T& a, const T& b) {return a<b? a=b, true:false;}
const int mod = 1e9+7, N = 2e5+5;
int msb(int val){return sizeof(int)*8-__builtin_clzll(val)-1;}
int n, m, k;
int a[N];
bool check(vector<int> &v){
	int n = v.size();
	int t = v[0] + v[n-1];
	for(int l=0,r=n-1;l<r;l++,r--){
		if(v[l] + v[r] != t)return false;
	}return true;
}
void solve(int test_case){
	int i, j;
	cin >> n >> k;
	m = n+k;
	for(i=1;i<=m;i++){
		cin >> a[i];
	}
	sort(a+1,a+m+1);
	if(k == 1){
		bool ok = true;
		int t = (a[1] + a[m-1]);
		for(int l=1,r=m-1;l<r;l++,r--){
			if(a[l] + a[r] != t)ok = false;
		}
		if(ok){
			for(i=1;i<=m-1;i++)cout << a[i] << ' ' ;
			cout << '\n';
			return;
		}
		ok = true;
		t = (a[2] + a[m]);
		for(int l=2,r=m;l<r;l++,r--){
			if(a[l] + a[r] != t)ok = false;
		}
		
		if(ok){
			for(i=2;i<=m;i++)cout << a[i] << ' ' ;
			cout << '\n';
			return;
		}
		t = a[1] + a[m];
		for(int l=2,r=m-1;l<r;l++,r--){
			if(a[l] + a[r] != t){
				vector<int> v;
				for(i=1;i<=m;i++){
					if(i==l)continue;
					v.pb(a[i]);
				}
				if(check(v)){
					for(auto x : v)cout << x << ' ' ;
					cout << '\n';
					return;
				}
				v.clear();
				for(i=1;i<=m;i++){
					if(i==r)continue;
					v.pb(a[i]);
				}
				if(check(v)){
					for(auto x : v)cout << x << ' ' ;
					cout << '\n';
					return;
				}
				assert(false);
			}
		}
	}else {
		set<int> possible_sums;
		for(i=1;i<=m;i++){
			for(j=i+1;j<=m;j++){
				possible_sums.insert(a[i] + a[j]);
			}
		}
		//~ cout << possible_sums.size() << '\n';
		for(auto sum : possible_sums){
			//~ cout << sum << '\n';
			map<int,int>cnt;
			for(i=1;i<=m;i++){
				cnt[a[i]]++;
			}
			int have = 0;
			vector<int> v;
			for(i=1;i<=m;i++){
				if(cnt[a[i]] <= 0)continue;
				if(a[i] >= sum)continue;
				if(sum-a[i] == a[i]){
					if(cnt[sum - a[i]] > 1){
						cnt[sum - a[i]]--;
						cnt[a[i]]--;
						have++;
						v.pb(a[i]);
						v.pb(sum-a[i]);
					}
				}else if(cnt[sum - a[i]] > 0){
					cnt[sum - a[i]]--;
					cnt[a[i]]--;
					have++;
					v.pb(a[i]);
					v.pb(sum-a[i]);
				}
			}
			//~ cout << sum << ' ' << have << '\n';
			if(have >= n/2){
				sort(all(v));
				for(auto x : v)cout << x << ' ' ;
				cout << '\n';
				return;
			}
		}
	}
	return;
}
/*

4 1
2 4 8 6 10

*/
signed main(){
	FASTIO;
//~ #define MULTITEST 1
#if MULTITEST
	int _T;
	cin >> _T;
	for(int T_CASE = 1; T_CASE <= _T; T_CASE++)
		solve(T_CASE);
#else
	solve(1);
#endif
	return 0;
}

Subtask #1
11.0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 364 KB Output is correct

Subtask #2
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 5 ms 1008 KB Output is correct
2 Correct 38 ms 4460 KB Output is correct
3 Correct 37 ms 4460 KB Output is correct
4 Correct 36 ms 4460 KB Output is correct

Subtask #3
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 3137 ms 622304 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 748 KB Output is correct
2 Correct 497 ms 1132 KB Output is correct
3 Correct 7 ms 1132 KB Output is correct
4 Correct 177 ms 1260 KB Output is correct
5 Correct 8 ms 1132 KB Output is correct

Subtask #5
9.0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 364 KB Output is correct
2 Correct 1 ms 364 KB Output is correct
3 Correct 1 ms 364 KB Output is correct

Subtask #6
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 364 KB Output is correct
2 Execution timed out 3093 ms 2860 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #7
0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 364 KB Output is correct
2 Execution timed out 3088 ms 335556 KB Time limit exceeded
3 Halted 0 ms 0 KB -

Subtask #8
0 / 13.0

# 결과 실행 시간 메모리 Grader output
1 Correct 24 ms 4716 KB Output is correct
2 Execution timed out 3115 ms 506688 KB Time limit exceeded
3 Halted 0 ms 0 KB -