Submission #1318853

#TimeUsernameProblemLanguageResultExecution timeMemory
1318853husuuuTable Tennis (info1cup20_tabletennis)C++20
34 / 100
101 ms17344 KiB
#include <bits/stdc++.h>
using namespace std;
///////////////////////////////////////////////
//#define int long long
#define endl "\n"
#define IO ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define INF 0x3F3F3F3F3F3F3F3F
#define ss second
#define ff first
#define pb push_back
#define ins insert
#define fori for(int i = 0 ; i < n ; i ++)
///////////////////////////////////////////////
const int sz = 2e3 + 5 ;
const int LG = 20 ;
const int N = 1e3 ;
const int mod = 1e9 + 7 ;
const int MAXM = 1e5 ;
///////////////////////////////////////////////
///////////////////////////////////////////////
void solve() {
  int n ;
  cin >> n ;
  int nc = n ;
  int k ;
  cin >> k ;
  n += k ; 
  vector<int>a(n) ;
  map<int , int>mp ;
  map<int , int>pos ;
  for(int i = 0 ; i < n ; i ++) {
    cin >> a[i] ;
    mp[a[i]] ++ ;
  }
  if(nc == 2 and k == 1) {
    cout << a[0] << ' ' << a[1] ;
    return ;
  }
  sort(a.begin() , a.end()) ;
  if(k == 1) {
    for(int i = 0 ; i < n ; i ++) {
      pos[a[i]] = i ;
    }
    int nn = a.size() ;
    vector<bool>check(nn , 0) ;
    int l = 1 ;
    int r = nn - 1;
    map<int , int>rep ;
    while(l < r) {
      int sum = a[l] + a[r] ;
      l ++ ;
      r -- ;
      rep[sum] ++ ;
    }
    l = 0 ;
    r = nn - 2 ;
    while(l < r) {
      int sum = a[l] + a[r] ;
      l ++ ;
      r -- ;
      rep[sum] ++ ;
    }
    l = 0 ; 
    r = nn - 1 ;
    while(l < r) {
      int sum = a[l] + a[r] ;
      l ++ ; 
      r -- ;
      rep[sum] ++ ;
    }
    int sum = 0 ;
    int maks = -1 ;
    for(auto i : rep) {
      if(maks < i.ss) {
        maks = i.ss ;
        sum = i.ff ;
      }
    }
    //cout << sum << endl ;
    int ind = -1 ;
    for(int i = 0 ; i < n ; i ++) {
      if(check[i] == 0) {
        check[i] = 1 ;
        int dif = sum - a[i] ;
        //cout << dif << ' ' ;
        if(mp[dif] == 0) {
          ind = i ;
          break ;
        }
        if(mp[dif] == 1 and dif == a[i]) {
          ind = i ;
          break ;
        }
        check[pos[dif]] = 1 ;
      }
    }
    //cout << endl ;
    for(int i = 0 ; i < n ; i ++) {
      if(i != ind) {
        cout << a[i] << ' ' ;
      }
    }
  }
  else {
    if(k == 2) {
      for(int i = 0 ; i < n ; i ++) {
        pos[a[i]] = i ;
      }
      int nn = a.size() ;
      vector<bool>check(nn , 0) ;
      int l = 1 ;
      int r = nn - 1;
      map<int , int>rep ;
      l = 0 ;
      r = nn - 3 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      l = 1 ;
      r = nn - 2 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      l = 2 ;
      r = nn - 1 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      int sum = 0 ;
      int maks = -1 ;
      for(auto i : rep) {
        if(maks < i.ss) {
          maks = i.ss ;
          sum = i.ff ;
        }
      }
      //cout << sum << endl ;
      int ind = -1 ;
      int ind1 = -1 ;
      vector<int>ans ;
      for(int i = 0 ; i < n ; i ++) {
        int dif = sum - a[i] ;
        //cout << dif << ' ' ;
        if(mp[dif] == 0) {
        }
        else if(mp[dif] == 1 and dif == a[i]) {
        }
        else {
          ans.push_back(a[i]) ;
        }
      }
      //cout << endl << ind1 << endl ;
      //cout << ind << endl ;
      //cout << sum << endl ;
      sort(ans.begin() , ans.end()) ;
      for(int i = 0 ; i < ans.size() ; i ++) {
        cout << ans[i] << ' ' ; 
      }
    }
    else if(k == 3) {
      for(int i = 0 ; i < n ; i ++) {
        pos[a[i]] = i ;
      }
      int nn = a.size() ;
      vector<bool>check(nn , 0) ;
      map<int , int>rep ;
      int l = 0 ;
      int r = nn - 4 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      l = 1 ;
      r = nn - 3 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      l = 2 ;
      r = nn - 2 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      l = 3 ;
      r = nn - 1 ;
      while(l < r) {
        int sum = a[l] + a[r] ;
        l ++ ;
        r -- ;
        rep[sum] ++ ;
      }
      int sum = 0 ;
      int maks = -1 ;
      for(auto i : rep) {
        if(maks < i.ss) {
          maks = i.ss ;
          sum = i.ff ;
        }
      }
      vector<int>ans ;
      for(int i = 0 ; i < n ; i ++) {
        int dif = sum - a[i] ;
        if(mp[dif] == 0) {
        }
        else if(mp[dif] == 1 and dif == a[i]) {
        }
        else {
          ans.push_back(a[i]) ;
        }
      }
      sort(ans.begin() , ans.end()) ;
      for(int i = 0 ; i < ans.size() ; i ++) {
        cout << ans[i] << ' ' ; 
      }
    }
    else {
      for(int i = 0 ; i < n ; i ++) {
        pos[a[i]] = i ;
      }
      int nn = a.size() ;
      vector<bool>check(nn , 0) ;
      map<int , int>rep ;
      for(int i = 0 ; i <= k ; i ++) {
        int l = i ;
        int r = nn - i - 1 ;
        while(l < r) {
          int sum = a[l] + a[r] ;
          l ++ ;
          r -- ; 
          rep[sum] ++ ;
        }
      }
      int sum = 0 ;
      int maks = -1 ;
      for(auto i : rep) {
        if(maks < i.ss) {
          maks = i.ss ;
          sum = i.ff ;
        }
      }
      vector<int>ans ;
      for(int i = 0 ; i < n ; i ++) {
        int dif = sum - a[i] ;
        if(mp[dif] == 0) {
        }
        else if(mp[dif] == 1 and dif == a[i]) {
        }
        else {
          ans.push_back(a[i]) ;
        }
      }
      sort(ans.begin() , ans.end()) ;
      for(int i = 0 ; i < ans.size() ; i ++) {
        cout << ans[i] << ' ' ; 
      }
    }
  }
}
///////////////////////////////////////////////
///////////////////////////////////////////////
signed main() {
  IO ;
  int t = 1 ; // cin >> t ;
  while(t --) {
    solve() ;
    cout << endl ;
  }
}
#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...