Submission #794695

#	Time	Username	Problem	Language	Result	Execution time	Memory
794695		finn__	Dungeons Game (IOI21_dungeons)	C++17	89 / 100	7132 ms	1423412 KiB

dungeons

#include <bits/stdc++.h>

#include "dungeons.h"

using namespace std;

constexpr size_t N = 400001, K = 25, L = 9;

size_t n;
uint32_t f[L][N][K], m[L][N][K];
uint64_t u[L][N][K];
vector<int> s, p, w, l;

void init(int n_, vector<int> s_, vector<int> p_, vector<int> w_, vector<int> l_)
{
    n = n_;
    s = move(s_);
    p = move(p_);
    w = move(w_);
    l = move(l_);

    for (size_t k = 0; k < L; ++k)
    {
        for (size_t i = 0; i < n; ++i)
        {
            if (s[i] < 1ULL << (k * 3))
                f[k][i][0] = w[i], u[k][i][0] = s[i], m[k][i][0] = UINT32_MAX;
            else
                f[k][i][0] = l[i], u[k][i][0] = p[i], m[k][i][0] = s[i];
        }
        f[k][n][0] = n;
        u[k][n][0] = 0;
        m[k][n][0] = UINT32_MAX;
        for (size_t j = 1; j < K; ++j)
            for (size_t i = 0; i < n; ++i)
            {
                size_t const intermediate = f[k][i][j - 1];
                f[k][i][j] = f[k][intermediate][j - 1];
                u[k][i][j] = u[k][i][j - 1] + u[k][intermediate][j - 1];

                m[k][i][j] = m[k][i][j - 1];
                if (m[k][intermediate][j - 1] != UINT32_MAX)
                {
                    uint32_t v = m[k][intermediate][j - 1] - min<uint64_t>(m[k][intermediate][j - 1],
                                                                           u[k][i][j - 1]);
                    m[k][i][j] = min(m[k][i][j], v);
                }
            }
    }
}

long long simulate(int x, int z)
{
    long long strength = z;

    while (x != n)
    {
        if (strength >= s[x])
        {
            strength += s[x];
            x = w[x];
        }
        else
        {
            strength += p[x];
            x = l[x];
        }
        size_t const k = min<size_t>(L - 1, __lg(strength) / 3);

#pragma GCC unroll K
        for (size_t j = K - 1; j < K; --j)
            if (m[k][x][j] > strength) [[likely]]
            {
                strength += u[k][x][j];
                x = f[k][x][j];
                break;
            }
    }

    return strength;
}

Compilation message (stderr)

dungeons.cpp: In function 'void init(int, std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
dungeons.cpp:26:22: warning: comparison of integer expressions of different signedness: '__gnu_cxx::__alloc_traits<std::allocator<int>, int>::value_type' {aka 'int'} and 'long long unsigned int' [-Wsign-compare]
   26 |             if (s[i] < 1ULL << (k * 3))
dungeons.cpp: In function 'long long int simulate(int, int)':
dungeons.cpp:56:14: warning: comparison of integer expressions of different signedness: 'int' and 'size_t' {aka 'long unsigned int'} [-Wsign-compare]
   56 |     while (x != n)
      |            ~~^~~~

• Subtask 1: 11 / 11
• Subtask 2: 26 / 26
• Subtask 3: 13 / 13
• Subtask 4: 12 / 12
• Subtask 5: 27 / 27
• Subtask 6: 0 / 11