Submission #979667

#	Time	Username	Problem	Language	Result	Execution time	Memory
979667		canadavid1	A Plus B (IOI23_aplusb)	C++17	10 / 100	1 ms	348 KiB

aplusb

#include "aplusb.h"
#include <set>
#include <utility>
#include <tuple>
std::vector<int> smallest_sums(int N, std::vector<int> A, std::vector<int> B) {
  auto cmp = [&](auto a,auto b){return std::make_tuple(A[a.first]+B[a.second],a,b)<std::make_tuple(A[b.first]+B[b.second],a,b);};
  std::set<std::pair<int,int>,decltype(cmp)> q(cmp);
  std::vector<int> out;
  q.emplace(0,0);
  while(out.size()<N)
  {
    auto[a,b] = *q.begin();
    out.push_back(A[a]+B[b]);
    q.erase(q.begin());
    if(a+1<A.size()) q.emplace(a+1,b);
    if(b+1<B.size()) q.emplace(a,b+1);
  }
  return out;
}

Compilation message (stderr)

aplusb.cpp: In function 'std::vector<int> smallest_sums(int, std::vector<int>, std::vector<int>)':
aplusb.cpp:10:19: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   10 |   while(out.size()<N)
      |         ~~~~~~~~~~^~
aplusb.cpp:15:11: warning: comparison of integer expressions of different signedness: 'std::tuple_element<0, std::pair<int, int> >::type' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |     if(a+1<A.size()) q.emplace(a+1,b);
      |        ~~~^~~~~~~~~
aplusb.cpp:16:11: warning: comparison of integer expressions of different signedness: 'std::tuple_element<1, std::pair<int, int> >::type' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   16 |     if(b+1<B.size()) q.emplace(a,b+1);
      |        ~~~^~~~~~~~~

• Subtask 1: 10 / 10
• Subtask 2: 0 / 20
• Subtask 3: 0 / 30
• Subtask 4: 0 / 40