Submission #1346543

#TimeUsernameProblemLanguageResultExecution timeMemory
1346543bakhtiyarnTable Tennis (info1cup20_tabletennis)C++20
0 / 100
3104 ms384568 KiB
#include <bits/stdc++.h>
using namespace std;

const int N = 3e5+5;
int a[N];
map<int, int> best;

void build(int l, int r, int sm){
  while(l < r){
    if(a[l] + a[r] > sm) r--;
    else if(a[l]+a[r] < sm) l++;
    else {
      best[sm]++;
      l++;
      r--;
    }
  }
}

// for(int i=1; i<=n; i++) 
void solve(){
  int n, k; cin >> n >> k;
  for(int i=1; i<=n+k; i++) cin >> a[i];
  
  
  
  for(int r=n-2; r<=n+k; r++){
    int l = k+1;
    if(l >= r) continue;
    if(r-l+1 < n) continue;
    build(l, r, a[l]+a[r]);
  }
  
  int mx = 0, sm = 0;
  for(int l=k; l>=1; l--){
    for(int r=l+1; r<=n+k; r++){
      best[a[l]+a[r]]++;
      if(mx < best[a[l]+a[r]]) mx = best[a[l]+a[r]], sm = a[l]+a[r];
    }
  }
  
  map<int, int> pos;
  for(int i=1; i<=n+k; i++) pos[a[i]] = i;
  
  vector<int> ans;
  for(int i=1; i<=n+k and ans.size() < n; i++){
    if(pos[sm-a[i]]) {
      ans.push_back(a[i]);
      ans.push_back(sm-a[i]);
    }
  }
  sort(ans.begin(), ans.end());
  for(int i: ans) cout << i << ' ';
}

signed main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  solve();
}
#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...