제출 #852254

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
852254		alex_2008	던전 (IOI21_dungeons)	C++17	100 / 100	2621 ms	985296 KiB

dungeons

#include "dungeons.h"
#include <bits/stdc++.h>
#define mindiff first
#define pos second.first
#define strength second.second
using namespace std;
const long long N = 4e5 + 10;
const long long M = 10000001;
const int INF = 1e7;
pair<int, pair<int, int>> dp[N][8][24];
long long NN;
vector <long long> ss, pp, ww, ll;
long long nxt[N];
int lg8[M];
bool ch = true;
void init(int n, vector<int> s, vector<int> p, vector<int> w, vector<int> l) {
  	lg8[1] = 0;
  	for (int i = 1; i < 8; i++) {
  		lg8[i] = 0;
  	}
  	for (int i = 8; i < M; i++) {
		lg8[i] = lg8[i / 8] + 1;
    } 
  	NN = n;
	s.push_back(0);
	p.push_back(0);
	w.push_back(n);
	l.push_back(n);
	for (int i = 0; i <= n; i++) {
		ss.push_back(s[i]);
		if (w[i] != l[i]) ch = false;
    }
	nxt[n] = 0;
	for (int i = n - 1; i >= 0; i--) {
		nxt[i] = s[i] + nxt[w[i]];
	}
	for (int i = 0; i <= n; i++) {
		pp.push_back(p[i]);
	}
	for (int i = 0; i <= n; i++) {
		ww.push_back(w[i]);
	}
	for (int i = 0; i <= n; i++) {
		ll.push_back(l[i]);
	}
	for (int k = 0; k < 24; k++) {
		for (int i = 0; i <= n; i++) {
			for (int j = 0; j < 8; j++) {
				if (k == 0) {
					dp[i][j][k].pos = ((1 << (3 * j)) >= s[i]) ? w[i] : l[i];
					dp[i][j][k].strength = ((1 << (3 * j)) >= s[i]) ? s[i] : p[i];
					dp[i][j][k].mindiff = ((1 << (3 * j)) < s[i]) ? s[i] : INF;
				}
				else {
					int p = dp[i][j][k - 1].pos;
					dp[i][j][k].pos = dp[p][j][k - 1].pos;
					dp[i][j][k].strength = min(INF, dp[i][j][k - 1].strength + dp[p][j][k - 1].strength);
					dp[i][j][k].mindiff = max(0, min(dp[i][j][k - 1].mindiff, dp[p][j][k - 1].mindiff - dp[i][j][k - 1].strength));
				}
			}
		}
	}
}
long long simulate(int xx, int zz) {
	long long x = xx, z = zz;
	while (1) {
		if (z >= 10000000) {
			z += nxt[x];
			return z;
		}
		int v = (int)lg8[z];
		for (int k = 23; k >= 0; k--) {
			if (dp[x][v][k].pos == N || dp[x][v][k].mindiff <= z || (dp[x][v][k].strength + z >= 10000000) || lg8[dp[x][v][k].strength + z] > v) continue;
			z += dp[x][v][k].strength;
			x = dp[x][v][k].pos;
		}
		if (z >= ss[x]) {
			z += ss[x];
			x = ww[x];
		}
		else {
			z += pp[x];
			x = ll[x];
		}
		if (x == NN) return z;
	}
	return 0;
}
//int main() {
//	init(3, { 2, 6, 9 }, { 3, 1, 2 }, { 2, 2, 3 }, { 1, 0, 1 });
//	cout << simulate(0, 1) << "\n";
//	cout << simulate(2, 3) << "\n";
//}

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