제출 #1203270

#제출 시각아이디문제언어결과실행 시간메모리
1203270ArdelionA Plus B (IOI23_aplusb)C++20
100 / 100
78 ms9980 KiB
#include "aplusb.h"
#include <bits/stdc++.h>

using namespace std;
 
#define threesum cin.tie(NULL); cout.tie(NULL); ios_base::sync_with_stdio(false)
#define all(a) a.begin(), a.end()
#define F first
#define S second
//#define int long long
#define pii pair<int, int>
#define ppp pair<int, pii>
#define dout cout << fixed << setprecision(15)
#define mid ((l + r) / 2)
#define lc (2 * id)
#define rc (lc + 1)

vector<int> smallest_sums(int N, vector<int> A, vector<int> B) {
  //return A;
  sort(all(A));
  sort(all(B));
  map<pii, bool> seen;
  priority_queue<ppp> pq;
  pq.push({A[0] + B[0], {0, 0}});
  vector<int> ans;
  while((int)ans.size() < N){
    auto [dst, pr] = pq.top();
    auto [i, j] = pr;
    pq.pop();

    if(seen[{i, j}])
      continue;
    seen[{i, j}] = 1;

    ans.push_back(A[i] + B[j]);

    if(i + 1 < N)
      pq.push({-(A[i + 1] + B[j]), {i + 1, j}});
    if(j + 1 < N)
      pq.push({-(A[i] + B[j + 1]), {i, j + 1}});
  }
  return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...