# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1252805 | Giselus | Souvenirs (IOI25_souvenirs) | C++20 | 0 ms | 0 KiB |
#include <bits/stdc++.h>
using namespace std;
#define fwd(i, a, n) for (int i = (a); i < (n); i++)
#define rep(i, n) fwd(i, 0, n)
#define all(X) X.begin(), X.end()
#define sz(X) int(size(X))
#define pb push_back
#define eb emplace_back
#define st first
#define nd second
using pii = pair<int, int>; using vi = vector<int>;
using ll = long long; using ld = long double;
#ifdef LOC
auto SS = signal(6, [](int) { *(int *)0 = 0; });
#define DTP(x, y) auto operator << (auto &o, auto a) -> decltype(y, o) { o << "("; x; return o << ")"; }
DTP(o << a.st << ", " << a.nd, a.nd);
DTP(for (auto i : a) o << i << ", ", all(a));
void dump(auto... x) { (( cerr << x << ", " ), ...) << '\n'; }
#define deb(x...) cerr << setw(4) << __LINE__ << ":[" #x "]: ", dump(x)
#else
#define deb(...) 0
#endif
#define S 107
// pair<vi, ll> transaction(ll M);
ll P[S];
int how_many[S];
void f(int k, ll p) {
auto e = transaction(p);
for(auto v: e.st) how_many[v]++;
p -= e.nd;
if (e.st.size() == 1) {
P[k] = p;
return;
}
f(e.st[1], (p-1)/2);
}
void buy_souvenirs(int n, ll P0) {
P[0] = P0;
for(int i = 1; i <= n-1; i++) {
if(P[i])
continue;
f(i, P[i-1]-1);
}
for(int i = 1; i <= n-1; i++) {
while(how_many[i] < i) {
how_many[i++];
transaction(P[i]);
}
}
}
// int32_t main() {
// cin.tie(0)->sync_with_stdio(0);
// cout << fixed << setprecision(10);
// }