Submission #488152

#	Time	Username	Problem	Language	Result	Execution time	Memory
488152		8e7	던전 (IOI21_dungeons)	C++17	13 / 100	130 ms	47600 KiB

dungeons

//Challenge: Accepted
#include "dungeons.h"
#include <iostream>
#include <algorithm>
#include <vector>
#include <utility>
using namespace std;
void debug(){cout << endl;}
template<class T, class ... U> void debug(T a, U ... b){cout << a << " ", debug(b...);}
template<class T> void pary(T l, T r) {
	while (l != r) cout << *l << " ", l++;
	cout << endl;
}
#define ll long long
#define maxn 400005
#define pii pair<int, int>
#define ff first
#define ss second
#define io ios_base::sync_with_stdio(0);cin.tie(0);
int n;
vector<int> s, p, w, l;
struct WIN{
	int anc[19][maxn];
	ll ma[19][maxn];
	ll sum[19][maxn];
	void build() {
		for (int i = 0;i < n;i++) {
			anc[0][i] = w[i];
			ma[0][i] = s[i];
			sum[0][i] = s[i];
		}
		anc[0][n] = n;
		for (int i = 1;i < 19;i++) {
			for (int j = 0;j <= n;j++) {
				anc[i][j] = anc[i-1][anc[i-1][j]];
				sum[i][j] = sum[i-1][j] + sum[i-1][anc[i-1][j]];
				ma[i][j] = max(ma[i-1][j], ma[i-1][anc[i-1][j]] - sum[i-1][j]);
			}
		}
	}
	void move(int &x, ll &val) { //brings it to the nearest loss
		for (int i = 18;i >= 0;i--) {
			if (val >= ma[i][x]) {
				val += sum[i][x];
				x = anc[i][x];
			}
		}
	}
} win;
struct LOSE{
	int anc[25][maxn];
	ll mi[25][maxn];
	ll sum[25][maxn];
	void build() {
		for (int i = 0;i < n;i++) {
			anc[0][i] = l[i];
			mi[0][i] = s[i];
			sum[0][i] = p[i];
		}
		anc[0][n] = n;
		for (int i = 1;i < 25;i++) {
			for (int j = 0;j <= n;j++) {
				anc[i][j] = anc[i-1][anc[i-1][j]];
				sum[i][j] = sum[i-1][j] + sum[i-1][anc[i-1][j]];
				mi[i][j] = max((ll)0, min(mi[i-1][j], (ll)mi[i-1][anc[i-1][j]] - sum[i-1][j]));
			}
		}
	}
	void move(int &x, ll &val) { //brings it to the nearest win
		for (int i = 24;i >= 0;i--) {
			if (val < mi[i][x]) {
				val += sum[i][x];
				x = anc[i][x];
			}
		}
	}
} lose;
void init(int nv, vector<int> sv, vector<int> pv, vector<int> wv, vector<int> lv) {
	n = nv;
	s = sv, p = pv, w = wv, l = lv;	
	win.build();	
	lose.build();
}


long long simulate(int x, int z) {
	ll ans = z;
	lose.move(x, ans);
	win.move(x, ans);
	/*
	while (x != n) {
		win.move(x, ans);
		if (x == n) break;	
		ans += p[x];
		x = l[x];
	}
	*/
	return ans;
}
/*
3 2
2 6 9
3 1 2
2 2 3
1 0 1
0 1
2 3

3 3
14 14 14
2 1 1 
1 2 3
1 2 0
0 0
2 4
1 1
*/

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