Submission #439882

#TimeUsernameProblemLanguageResultExecution timeMemory
439882phathnvDungeons Game (IOI21_dungeons)C++17
11 / 100
1079 ms965356 KiB
#include "dungeons.h"
#include <bits/stdc++.h>

using namespace std;

const int N = 50007;//4e5 + 7;
const int L = 24;

int n;
vector<int> s, p, w, l;
int nxt[L][L][N];
long long f[L][L][N], sum[L][L][N];

void init(int _n, vector<int> _s, vector<int> _p, vector<int> _w, vector<int> _l) {
    n = _n;
    s = _s;
    p = _p;
    w = _w;
    l = _l;
    for (int b = 0; b < L; b++) {
        for (int i = 0; i < n; i++) {
            nxt[b][0][i] = (s[i] <= (1 << b)? w[i] : l[i]);
            sum[b][0][i] = (s[i] <= (1 << b)? s[i] : p[i]);
            f[b][0][i] = (s[i] < (1 << b)? 1e18 : s[i]);
        }
        nxt[b][0][n] = -1;
        for (int i = 1; i < L; i++)
            for (int j = 0; j <= n; j++) {
                nxt[b][i][j] = (nxt[b][i - 1][j] == -1? -1 : nxt[b][i - 1][nxt[b][i - 1][j]]);
                if (nxt[b][i][j] != -1) {
                    sum[b][i][j] = sum[b][i - 1][j] + sum[b][i - 1][nxt[b][i - 1][j]];
                    f[b][i][j] = min(f[b][i - 1][j], f[b][i - 1][nxt[b][i - 1][j]] - sum[b][i - 1][j]);
                }
            }
    }
	return;
}

long long simulate(int x, int _z) {
//    int tmp = x;
//    long long expec = _z;
//    while (tmp != n) {
//        if (expec >= s[tmp]) {
//            expec += s[tmp];
//            tmp = w[tmp];
//        } else {
//            expec += p[tmp];
//            tmp = l[tmp];
//        }
//    }
//    cerr << "expected: " << expec << ' ' << endl;
    long long z = _z;
    for (int b = 0; b < L; b++) {
        if (z >= (1 << (b + 1)))
            continue;
        for (int i = L - 1; i >= 0; i--) {
            if (nxt[b][i][x] == -1)
                continue;
            if (z < f[b][i][x]) {
                z += sum[b][i][x];
                x = nxt[b][i][x];
            }
        }
        if (x == n)
            return z;
        assert(z >= s[x]);
        z += s[x];
        x = w[x];
    }
    assert(x == n);
	return z;
}

#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...