Submission #685901

#	Time	Username	Problem	Language	Result	Execution time	Memory
685901		Alihan_8	Table Tennis (info1cup20_tabletennis)	C++17	9 / 100	24 ms	4280 KiB

tabletennis

#include <bits/stdc++.h>
// include <ext/pb_ds/assoc_container.hpp>
// include <ext/pb_ds/tree_policy.hpp>
// using namespace __gnu_pbds;
using namespace std;
#define all(x) x.begin(), x.end()
#define pb push_back
// define ordered_set tree<int,null_type,less_equal<int>,rb_tree_tag,tree_order_statistics_node_update>
#define mpr make pair
#define ln '\n'
void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
#define int long long
signed main(){
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int n, k; cin >> n >> k;
    vector <int> p(n+k);
    for ( auto &i: p ) cin >> i;
    const int S = 1 << n+k;
    for ( int mask = 0; mask < S; mask++ ){
        if ( __builtin_popcount(mask) != n ) continue;
        map <int,int> mp;
        vector <int> res;
        for ( int i = 0; i < n+k; i++ ){
            if ( mask >> i & 1 ) res.pb(p[i]), mp[p[i]]++;
        }
        int cnt = accumulate(all(res),  0ll), team = n/2;
        if ( cnt%team ) continue;
        cnt /= team;
        bool flag = true;
        for ( auto i: res ){
            if ( !mp.count(cnt-i) ){
                flag = false;
                break;
            }
            if ( --mp[cnt-i] == 0 ) mp.erase(cnt-i);
        }
        if ( flag ){
            sort(all(res));
            for ( auto i: res ) cout << i << ' ';
            return 0;
        }
    }
    assert(false);

    cout << '\n';
}

Compilation message (stderr)

tabletennis.cpp: In function 'int main()':
tabletennis.cpp:20:25: warning: suggest parentheses around '+' inside '<<' [-Wparentheses]
   20 |     const int S = 1 << n+k;
      |                        ~^~
tabletennis.cpp: In function 'void IO(std::string)':
tabletennis.cpp:11:29: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                      ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
tabletennis.cpp:11:70: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   11 | void IO(string name){freopen((name+".in").c_str(),"r",stdin); freopen((name+".out").c_str(),"w",stdout);}
      |                                                               ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 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