//#include "festival.h" #include<bits/stdc++.h> using namespace std; #define pb push_back #define ff first #define ss second #define ll long long std::vector<ll> max_coupons(ll A, std::vector<ll> P, std::vector<ll> T) { ll N=P.size(); vector<vector<pair<ll,ll>>> v(5); for(ll i=0;i<N;i++){ v[T[i]].pb({P[i],i}); } for(ll i=1;i<=2;i++){ sort(v[i].begin(),v[i].end()); } // ps? vector<ll>ps((ll)v[1].size()); if(v[1].size()>0){ ps[0]=v[1][0].ff; } for(ll i=1;i<v[1].size();i++){ ps[i]=ps[i-1]+(ll)v[1][i].ff; } pair<ll,ll>cnt={-1,-1}; cnt.ff= upper_bound(ps.begin(),ps.end(),(ll)A)-ps.begin(); for(ll i=0;i<v[2].size();i++){ if(A<v[2][i].ff)break; A-=v[2][i].ff; A*=2; ll h= upper_bound(ps.begin(),ps.end(),(ll)A)-ps.begin(); cnt=max(cnt,{i+1+h,i+1}); } vector<ll>ans; for(ll i=0;i<cnt.ss;i++){ ans.pb(v[2][i].ss); } for(ll i=0;i<cnt.ff-cnt.ss;i++){ ans.pb(v[1][i].ss); } return ans; }

/usr/bin/ld: /tmp/ccT3Gbfj.o: in function `main':
grader.cpp:(.text.startup+0x22a): undefined reference to `max_coupons(int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >)'
collect2: error: ld returned 1 exit status