#include "dungeons.h"
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define se second
const int LOG=30;
vector<int> s, p, w, l, dist, values;
vector<vector<vector<pair<int, int>>>> up;
int n, m;
void init(signed _n, vector<signed> _s, vector<signed> _p, vector<signed> _w, vector<signed> _l) {
set<int> st; for (auto &u: _s) st.insert(u);
for (auto &u: st) values.pb(u);
n=_n, m=(int)values.size();
for (auto &u: _s) s.pb(u);
for (auto &u: _p) p.pb(u);
for (auto &u: _w) w.pb(u);
for (auto &u: _l) l.pb(u);
dist.resize(n+1); dist[n]=0;
for (int i=n-1; i>=0; i--) dist[i]=dist[w[i]]+s[i];
up.resize(m, vector<vector<pair<int, int>>>(n+1, vector<pair<int, int>>(LOG, {n, 0})));
for (int x=0; x<m; x++) {
for (int i=0; i<n; i++) {
up[x][i][0]={l[i], p[i]};
if (s[i]<values[x]) up[x][i][0]={w[i], s[i]};
}
for (int bit=1; bit<LOG; bit++) {
for (int i=0; i<n; i++) {
up[x][i][bit]=up[x][up[x][i][bit-1].fi][bit-1];
up[x][i][bit].se+=up[x][i][bit-1].se;
}
}
}
return;
}
int simulate(signed x, signed z) {
int ans=z;
for (int cur=0; cur<m; cur++) {
for (int bit=LOG-1; bit>=0; bit--) {
if (up[cur][x][bit].fi!=n && up[cur][x][bit].se+ans<values[cur]) ans+=up[cur][x][bit].se, x=up[cur][x][bit].fi;
}
if (ans<values[cur] && x!=n) ans+=up[cur][x][0].se, x=up[cur][x][0].fi;
}
return ans+dist[x];
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |