//#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<int> 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; }

festival.cpp: In function 'std::vector<int> max_coupons(long long int, std::vector<long long int>, std::vector<long long int>)':
festival.cpp:41:12: error: could not convert 'ans' from 'vector<long long int>' to 'vector<int>'
   41 |     return ans;
      |            ^~~
      |            |
      |            vector<long long int>