Submission #1252277

#	Time	Username	Problem	Language	Result	Execution time	Memory
1252277		tranvinhhuy2010	Souvenirs (IOI25_souvenirs)	C++20	100 / 100	12 ms	412 KiB

souvenirs

#include "souvenirs.h"
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

int n, last, cnt[105] = {};
ll p[105] = {};

ll adu_calculate(ll s, int k) {
    s += k*(k-1)/2 + k - 1;
    return s/k;
}

int adu_build(ll money) {
    auto [v, remain] = transaction(money);
    ll sum = money - remain;

    for (int i : v)
        cnt[i]++;

    while (v[0]<last) {
        while (v.back()>=last) {
            sum -= p[v.back()];
            v.pop_back();
        }

        if (v[0]+1==last) break;
        last = adu_build(adu_calculate(sum, v.size()) - 1);
    }

    p[v[0]] = sum;
    return v[0];
}

void adu_complete() {
    for (int i=1; i<n; i++) {
        while (cnt[i]<i) {
            transaction(p[i]);
            cnt[i]++;
        }
    }
}

void buy_souvenirs(int N, ll P0) {
    n = last = N;
    p[0] = P0;
    adu_build(P0-1);
    adu_complete();
}

• Subtask 0: 0 / 0
• Subtask 1: 4 / 4
• Subtask 2: 3 / 3
• Subtask 3: 14 / 14
• Subtask 4: 18 / 18
• Subtask 5: 28 / 28
• Subtask 6: 33 / 33