제출 #1330174

#제출 시각아이디문제언어결과실행 시간메모리
1330174somefolkFestival (IOI25_festival)C++20
0 / 100
166 ms139900 KiB
#include <bits/stdc++.h>
using namespace std;

#define int long long
const int INF = 1e16;

struct Coupon {
	int p, t, idx;
	int apply(int tk){
		tk -= p; tk *= t;
		if(tk > INF) tk = INF;
		if(tk < 0) tk = -1;
		return tk;
	}
};

vector<signed> max_coupons(signed a, vector<signed> p, vector<signed> t){
    int n = (int)p.size();

    deque<Coupon> nr, ones;
    for(int i = 0; i < n; i++){
        if(t[i] == 1) ones.push_back({p[i], t[i], i});
        else nr.push_back({p[i], t[i], i});
    }

    sort(nr.begin(), nr.end(), [&](auto left, auto right){
        if(left.t == 1 && right.t == 1) return left.p < right.p;
        return left.p * left.t * (right.t - 1) < right.p * right.t * (left.t - 1);
    });
    sort(nr.begin(), nr.end(), [&](auto left, auto right){
        if(left.t == 1 && right.t == 1) return left.p < right.p;
        return left.p * left.t * (right.t - 1) < right.p * right.t * (left.t - 1);
    });

    vector<signed> sol;

    while(!nr.empty()){
        Coupon curr = nr.back();
        if(curr.apply(a) < a) break;
        a = curr.apply(a);
        sol.push_back(curr.idx);
        nr.pop_back();
    }

    vector<vector<int>> dp((int)nr.size()+1, vector<int>(70, -1)); dp[0][0] = a;
    vector<vector<bool>> vis((int)nr.size()+1, vector<bool>(70, false));
    for(int i = 1; i <= (int)nr.size(); i++){
        for(int j = 0; j < 70; j++){
            dp[i][j] = dp[i-1][j];
        }
        Coupon curr = nr[i-1];
        for(int j = 1; j < 70; j++){
            if(dp[i-1][j-1] == -1) continue;
            if(curr.apply(dp[i-1][j-1]) > dp[i][j]){
                dp[i][j] = curr.apply(dp[i-1][j-1]);
                vis[i][j] = true;
            }
        }
    }

    vector<int> pref(ones.size() + 1);
    pref[0] = 0;
    for(int i = 0; i < (int)ones.size(); i++){
        pref[i+1] = pref[i] + ones[i].p;
    }
    auto get = [&](int cnt){
        return (int)(upper_bound(pref.begin(), pref.end(), cnt) - pref.begin());
    };

    int mx = 0, mxIdx = 0;
    for(int i = 0; i < 70; i++){
        if(dp[nr.size()][i] == -1) continue;
        if(i + get(dp[nr.size()][i]) > mx){
            mx = get(dp[nr.size()][i]) + i;
            mxIdx = i;
        }
    }

    vector<int> sol2;
    int i = nr.size(), j = mxIdx;
    while(i > 0){
        if(j >= 0 && vis[i][j]){
            sol2.push_back(nr[i-1].idx);
            i--; j--;
        } else {
            i--;
        }
    }
    reverse(sol2.begin(), sol2.end());
    for(int i : sol2) sol.push_back(i);

    int res = dp[nr.size()][mxIdx];
    for(int i = 0; i < (int)ones.size(); i++){
        if(res < ones[i].p) break;
        res = ones[i].apply(res);
        sol.push_back(ones[i].idx);
    }

    return sol;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...