Submission #443228

#	Time	Username	Problem	Language	Result	Execution time	Memory
443228		hhhhaura	Dungeons Game (IOI21_dungeons)	C++17	50 / 100	7042 ms	540392 KiB

dungeons

#include "dungeons.h"
#define wiwihorz
#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#pragma loop-opt(on)

#define rep(i, a, b) for(int i = a; i <= b; i ++)
#define rrep(i, a, b) for(int i = b; i >= a; i--)
#define all(x) x.begin(), x.end()
#define ceil(a, b) ((a + b - 1) / (b))

#define INF 1000000000000000000
#define MOD 1000000007
#define eps (1e-9)

using namespace std;

#define ll long long int
#define lld long double
#define pii pair<int, int>
#define random mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count())

#ifdef wiwihorz
#define print(a...) cerr << "Line " << __LINE__ << ": ", kout("[" + string(#a) + "] = ", a)
void vprint(auto L, auto R) { while(L < R) cerr << *L << " \n"[next(L) == R], ++L;}
void kout() {cerr << endl;}
template<class T1, class ... T2> void kout(T1 a, T2 ... e) { cerr << a << " ", kout(e...);}
#else
#define print(...) 0
#define vprint(...) 0
#endif 
int N, lg, cs;
vector<vector<ll>> to, cost, need, tol, getl;
vector<int> S, P, W, L;
void init(int n, vector<int> s, vector<int> p, vector<int> w, vector<int> l) {
	N = n, lg = 32;
	S = s, P = p, W = w, L = l;
	to.assign(lg + 1, vector<ll>(n + 1, n));
	cost.assign(lg + 1, vector<ll>(n + 1, 0));
	need.assign(lg + 1, vector<ll>(n + 1, 0));
	tol.assign(lg + 1, vector<ll>(n + 1, n));
	getl.assign(lg + 1, vector<ll>(n + 1, 0));
	rep(i, 0, n - 1) {
		to[0][i] = w[i];
		cost[0][i] = s[i];
		need[0][i] = s[i];
		tol[0][i] = l[i];
		getl[0][i] = p[i];
	}
	rep(i, 1, lg) rep(j, 0, n) {
		to[i][j] = to[i - 1][to[i - 1][j]];
		cost[i][j] = cost[i - 1][j] + cost[i - 1][to[i - 1][j]];
		need[i][j] = max(need[i - 1][j], need[i - 1][to[i - 1][j]] - cost[i - 1][j]);
		tol[i][j] = tol[i - 1][tol[i - 1][j]];
		getl[i][j] = getl[i - 1][j] + getl[i - 1][tol[i - 1][j]];
	}
	cs = 1;
	rep(i, 0, n - 1) cs &= (s[i] == s[0]);
	return;
}
pair<int, ll> go(int x, ll z) {
	rrep(i, 0, lg) {
		if(need[i][x] <= z) {
			z = z + cost[i][x];
			x = to[i][x];
		}
	}
	return {x, z};
}
pair<int, ll> gol(int x, ll z) {
	rrep(i, 0, lg) {
		if(z + getl[i][x] < S[0]) {
			z = z + getl[i][x];
			x = tol[i][x];
		}
	}
	if(x != N) z = z + P[x], x = L[x];
	return {x, z};
}
ll simulate(int x, int z) {
	ll cc = z;
	while(x != N) {
		tie(x, cc) = go(x, cc);
		if(x != N && cs) tie(x, cc) = gol(x, cc); 
		else if(x != N && !cs) cc += P[x], x = L[x];
	}	
	return cc;
}

Compilation message (stderr)

dungeons.cpp:5: warning: ignoring '#pragma loop ' [-Wunknown-pragmas]
    5 | #pragma loop-opt(on)
      | 
dungeons.cpp:25:13: warning: use of 'auto' in parameter declaration only available with '-fconcepts-ts'
   25 | void vprint(auto L, auto R) { while(L < R) cerr << *L << " \n"[next(L) == R], ++L;}
      |             ^~~~
dungeons.cpp:25:21: warning: use of 'auto' in parameter declaration only available with '-fconcepts-ts'
   25 | void vprint(auto L, auto R) { while(L < R) cerr << *L << " \n"[next(L) == R], ++L;}
      |                     ^~~~

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