Submission #1144988

#TimeUsernameProblemLanguageResultExecution timeMemory
1144988MedetbekTable Tennis (info1cup20_tabletennis)C++20
87 / 100
3100 ms170480 KiB
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("Ofast,unroll-loops,fast-math,O3")
#include <bits/stdc++.h>

#define ll long long
#define int long long
#define all(v) v.begin(), v.end()
#define nl '\n'
#define pb push_back
#define sz(s) (int)(s).size()
#define f first
#define s second
#define boost ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);

using namespace std;
const ll N = 1e5+10, MX = 1e18+1, MOD = 1e9 + 7;
void solve(){
    ll n, k;
    cin >> n >> k;
    ll m = n + k;
    ll a[m+1];
    for(int i = 1; i <= m; i++){
        cin >> a[i];
    }
    sort(a + 1, a + 1 + m);
    map <int, int> cnt;
    vector <int> v;
    set <pair <int, int> > sum;
    for(int i = 1; i <= min(k+k+k, m); i++){
        for(int j = m; j >= max(m - (k+k+k), 1ll); j--){
            ll sm = a[i]+a[j];
            cnt[sm]++;
            v.pb(sm);
        }
    }
    for(auto x:v){
        if(cnt[x] > 0){
            sum.insert({cnt[x], x});
            cnt[x]=0;
        }
    }
    for(auto [y, x]:sum){
        ll l = 1, r = m; 
        vector <int> ans;
        while(l < r){
            if(a[l] + a[r] > x){
                r--;
            }else if(a[l] + a[r] < x){
                l++;
            }else{
                ans.pb(a[l]);
                ans.pb(a[r]);
                l++;
                r--;
            }
        }
        if(sz(ans) >= n){
            sort(all(ans));
            for(int i = 0; i < n; i++){
                cout << ans[i] << " ";
            }
            return;
        }
    }
}
main(){
    ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
    ll ql =1;
	 while(ql--){
	    solve();
	}
}

Compilation message (stderr)

tabletennis.cpp:66:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   66 | main(){
      | ^~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...