#include"dungeons.h"
#include<vector>
#include<iostream>
typedef long long ll;
using namespace std;
int n;
struct Dungeon{
vector<int>nxt1; // win
vector<int>nxt2; // lose
vector<ll>st1; // strength
vector<ll>st2; // gain on losing
vector<vector<int>>jmp; // where to jump?
vector<vector<ll>>gain; // strength gain of that jump
vector<vector<ll>>win; // strength requirement to win in that jump
int pw;
ll thres=0;
Dungeon(int n):nxt1(n),nxt2(n),st1(n),st2(n){}
Dungeon()=default;
void setup(){
jmp=vector<vector<int>>(n+1,vector<int>((pw+1)/2));
gain=vector<vector<ll>>(n+1,vector<ll>((pw+1)/2));
win=vector<vector<ll>>(n+1,vector<ll>((pw+1)/2));
for(int p=0;p<pw;p++)
jmp[n][p/2]=n;
for(int i=0;i<n;i++){
jmp[i][0]=nxt2[i];
gain[i][0]=st2[i];
win[i][0]=st1[i];
}
for(int p=1;p<pw;p++){
for(int i=0;i<n;i++){
gain[i][p/2]=gain[i][(p-1)/2]+gain[jmp[i][(p-1)/2]][(p-1)/2];
if(win[jmp[i][(p-1)/2]][(p-1)/2]==1e9)
win[i][p/2]=win[i][(p-1)/2];
else
win[i][p/2]=max(0LL,min(1LL*win[i][(p-1)/2],1LL*win[jmp[i][(p-1)/2]][(p-1)/2]-gain[i][(p-1)/2]));
}
for(int i=0;i<n;i++)
jmp[i][p/2]=jmp[jmp[i][(p-1)/2]][(p-1)/2];
}
}
// simulate until it either reaches n-th node, wins, or strength>=thres
pair<int,ll>simulate(int x,ll st){
for(int p=pw-1;p>=0;p--){
while(jmp[x][p/2]!=n&&(1LL*win[x][p/2]>st||win[x][p/2]==1e9)&&st+gain[x][p/2]<thres){
st+=gain[x][p];
x=jmp[x][p];
}
}
while(x!=n&&st<thres){
// do 1 more jump
if(st>=st1[x]&&st1[x]!=1e9){
st+=st1[x];
x=nxt1[x];
break;
}else{
st+=st2[x];
x=nxt2[x];
}
}
return{x,st};
}
};
Dungeon dun[25];
void init(int N,vector<int>s,vector<int>p,vector<int>w,vector<int>l){
n=N;
for(int pw=0;pw<25;pw++){
dun[pw]=Dungeon(n);
dun[pw].thres=2LL<<pw;
dun[pw].pw=pw+1;
for(int i=0;i<n;i++){
if(s[i]<=1<<pw){
dun[pw].nxt1[i]=0;
dun[pw].nxt2[i]=w[i];
dun[pw].st1[i]=1e9;
dun[pw].st2[i]=s[i];
}else if(s[i]>=2<<pw){
dun[pw].nxt1[i]=0;
dun[pw].nxt2[i]=l[i];
dun[pw].st1[i]=1e9;
dun[pw].st2[i]=p[i];
}else{
dun[pw].nxt1[i]=w[i];
dun[pw].nxt2[i]=l[i];
dun[pw].st1[i]=s[i];
dun[pw].st2[i]=p[i];
}
}
}
dun[24].thres=1e18;
for(int pw=0;pw<25;pw++)
dun[pw].setup();
}
ll simulate(int x,int z){
ll st=z;
for(int p=0;p<25;p++){
while(x!=n&&st<2LL<<p){
auto r=dun[p].simulate(x,st);
x=r.first;
st=r.second;
}
}
return st;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
7004 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
9 ms |
7180 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
9 ms |
7180 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
9 ms |
7180 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
7 ms |
7004 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |