| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 | 
|---|---|---|---|---|---|---|---|
| 824248 | esomer | 죄수들의 도전 (IOI22_prison) | C++17 | 0 ms | 0 KiB | 
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
vector<vector<int>> ans(23);
vector<vector<int>> inds(8);
int val(int n, int exp){
	vector<ll> pw(9);
	pw[0] = 1;
	for(int i = 1; i <= 8; i++) pw[i] = pw[i-1] * 3;
	for(int i = 8; i >= 0; i--){
		int which;
		if(n >= pw[i] * 2) {n -= pw[i] * 2; which = 2;}
		else if(n >= pw[i]) {n -= pw[i]; which = 1;}
		else{
			which = 0;
		}
		if(i == exp) return which;
	}
	return 0;
}
void build(int ind, int exp, int should){
	if(exp == 0){
		for(int i = 1; i <= N; i++){
			int which = val(i, exp);
			if(which != should){
				if(which > should) ans[ind][i] = -(ans[ind][0] + 1);
				else ans[ind][i] = -((1 - ans[ind][0]) + 1);
			}
		}
		return;
	}
	if(exp == 1){
		//Optimize 2 queries.
		for(int i : inds[exp - 1]){
			ans[i][0] = 1 - ans[ind][0];
		}
		for(int i = 1; i <= N; i++){
			int which = val(i, exp);
			if(which != should){
				if(which > should) ans[ind][i] = -(ans[ind][0] + 1);
				else ans[ind][i] = -((1 - ans[ind][0]) + 1);
			}else{
				which = val(i, exp - 1);
				if(which == 2){
					ans[ind][i] = -(ans[ind][0] + 1);
				}else if(which == 0){
					ans[ind][i] = -((1 - ans[ind][0]) + 1);
				}else{
					ans[ind][i] = inds[exp-1][0];
				}
			}
		}
		build(inds[exp - 1][0], exp - 1, 1);
		return;
	}
	for(int i : inds[exp - 1]){
		ans[i][0] = 1 - ans[ind][0];
	}
	for(int i = 1; i <= N; i++){
		int which = val(i, exp);
		if(which != should){
			if(which > should) ans[ind][i] = -(ans[ind][0] + 1);
			else ans[ind][i] = -((1 - ans[ind][0]) + 1);
		}else{
			which = val(i, exp - 1);
			ans[ind][i] = inds[exp-1][which];
		}
	}
	build(inds[exp - 1][0], exp - 1, 0);
	build(inds[exp - 1][1], exp - 1, 1);
	build(inds[exp - 1][2], exp - 1, 2);
}
vector<vector<int>> devise_strategy(int N) {
	ans[0][0] = 0;
	int curr = 1;
	for(int i = 7; i >= 1; i--){
		inds[i].push_back(curr); curr++;
		inds[i].push_back(curr); curr++;
		inds[i].push_back(curr); curr++;
	}
	inds[0].push_back(curr);
	build(0, 8, 0);
	return ans;
}
