| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1316311 | ezzzay | 축제 (IOI25_festival) | C++20 | 0 ms | 0 KiB |
//#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;
}