제출 #1316310

#제출 시각아이디문제언어결과실행 시간메모리
1316310ezzzay축제 (IOI25_festival)C++20
컴파일 에러
0 ms0 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<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;
}

컴파일 시 표준 에러 (stderr) 메시지

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>