# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
840103 | Faisal_Saqib | A Plus B (IOI23_aplusb) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
vector<int> smallest_sum(int n,vector<int> a,vector<int> b)
{
vector<int> ans;
int can[a.size()];
int m=b.size();
for(int i=0;i<n;i++)
{
can[i]=0;
}
set<pair<int,pair<int,int>>> pq;
pq.insert({a[0]+b[0],{0,0}});
while(ans.size()<(n) and pq.size()>0)
{
auto p=*begin(pq);
pq.erase(begin(pq));
ans.push_back(p.first);
if(p.second.first< (n-1)){
pq.insert({a[p.second.first+1]+b[can[p.second.first+1]],{p.second.first+1,can[p.second.first+1]}});
}
if(p.second.second< (m-1))
{
pq.insert({a[p.second.first]+b[p.second.second+1],{p.second.first,p.second.second+1}});
can[p.second.first]+=1;
}
}
for(auto i:ans)
{
cout<<i<<' ';
}
cout<<endl;
return ans;
}
int main()
{
int n;
cin>>n;
vector<int> a,b;
for(int i=0;i<n;i++)
{
int x;
cin>>x;
a.push_back(x);
}
for(int i=0;i<n;i++)
{
int x;
cin>>x;
b.push_back(x);
}
smallest_sum(n,a,b);
}