Submission #1060710

#	Time	Username	Problem	Language	Result	Execution time	Memory
1060710		SzymonKrzywda	A Plus B (IOI23_aplusb)	C++17	100 / 100	84 ms	14344 KiB

aplusb

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


vector<int> smallest_sums(int n, vector<int> A, vector<int> B){
    
    priority_queue<pair<int,pair<int,int>>> kolejka;
    set<pair<int,int>> secik;
    
    vector<int> wynik(n);
    
    kolejka.push({-(A[0]+B[0]),{0,0}});
    
    for (int i=0; i<n; i++){
        wynik[i] = -kolejka.top().first;
        
        int y = kolejka.top().second.first;
        int x = kolejka.top().second.second;
        kolejka.pop();
        
        if (y+1<n && secik.find({y+1,x})==secik.end()){
            kolejka.push({-(A[y+1]+B[x]),{y+1,x}});
            secik.insert({y+1,x});
        } 
        if (x+1<n && secik.find({y,x+1})==secik.end()){
            kolejka.push({-(A[y]+B[x+1]),{y,x+1}});
            secik.insert({y,x+1});
        } 
    
    }
    
    return wynik;
    
}

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