제출 #1346513

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1346513		bakhtiyarn	Table Tennis (info1cup20_tabletennis)	C++20	72 / 100	3102 ms	335060 KiB

tabletennis

#include <bits/stdc++.h>
using namespace std;
#define int long long

const int N = 3e5+5;
unordered_map<int, int> pos;
int a[N];


vector<int> build(int l, int r, int sm, int k){
  vector<int> ans;
  for(int i=l; i<=r; i++){
    if(pos[sm-a[i]]) ans.push_back(a[i]);
  }
  return ans;
}




// 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 i=1; i<=n+k; i++) pos[a[i]] = i;
  
  for(int l=1; l<=2+k; l++){
    for(int r=n-2; r<=n+k; r++){
      if(l >= r) continue;
      int sm = a[l] + a[r];
      if(l-1 + n+k-r > k) continue;
      // 
        vector<int> ans = build(l, r, sm, k - (l-1 + n+k-r));
        if(ans.size() == n) {
          for(int i: ans) cout << i << " ";
          return;
        }
      // 
    }
  }
}

signed main() {
  ios_base::sync_with_stdio(0);
  cin.tie(0);
  cout.tie(0);
  solve();
}

• Subtask 1: 11 / 11
• Subtask 2: 9 / 9
• Subtask 3: 14 / 14
• Subtask 4: 15 / 15
• Subtask 5: 9 / 9
• Subtask 6: 14 / 14
• Subtask 7: 0 / 15
• Subtask 8: 0 / 13