# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
923409 | shoryu386 | Speedrun (RMI21_speedrun) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "speedrun.h"
#include <bits/stdc++.h>
using namespace std;
bool vis[1007];
int parset[1007];
vector<int> adj[1007];
vector<int> dorder[1007];
int n;
void dfs(int x, int p, int d){
vis[x] = 1;
parset[x] = p;
dorder[d].push_back(x);
for (auto y : adj[x]){
if (!vis[y]){
dfs(y, x, d+1);
}
}
}
void assignHints(int subtask, int N, int A[], int B[]) { /* your solution here */
setHintLen(20);
for (int x = 1; x < N; x++){
adj[A[x]].push_back(B[x]);
adj[B[x]].push_back(A[x]);
}
//let root be 1 always
dfs(1, 0, 0);
vector<int> inorder;
for (int x = 0; x <= N; x++){
for (auto y : dorder[x]){
inorder.push_back(y);
}
}
//data 1 is par
//data 2 is next
int next[N+1];
for (int x = 0; x < N-1; x++){
next[ inorder[x] ] = inorder[x+1];
}
next[inorder[N-1]] = inorder[0];
//encryption
for (int x = 1; x <= N; x++){
for (int z = 0; z <= 9; z++){
if (parset[x] & (1<<z)){
setHint(x, z+1, 1);
}
}
for (int z = 0; z <= 9; z++){
if (next[x] & (1<<z)){
setHint(x, z+11, 1);
}
}
}
return;
}
int par[1007];
int nextt[1007];
int cur;
#define WEIRD -23145
int n;
inline void setDetails(){
if (nextt[cur] != WEIRD) return;
par[cur] = 0; nextt[cur] = 0;
for (int z = 0; z <= 9; z++){
if (getHint(z+1)){
par[cur] += (1<<z);
}
}
for (int z = 0; z <= 9; z++){
if (getHint(z+11)){
nextt[cur] += (1<<z);
}
}
}
inline void assertGoTo(int x){
goTo(x);
}
inline void returnToRoot(){
setDetails();
int steps = 0;
while (cur != 1){
assertGoTo(par[cur]);
cur = par[cur];
setDetails();
assert(cur >= 1 && cur <= n);
}
}
vector<int> moves[1007];
inline void trace(int x){
returnToRoot();
for (auto z : moves[x]){
assertGoTo(z);
cur = z;
}
return;
}
void speedrun(int subtask, int N, int start) { /* your solution here */
cur = start;
n = N;
for (int x = 0; x < 1007; x++) par[x] = WEIRD;
for (int x = 0; x < 1007; x++) nextt[x] = WEIRD;
returnToRoot();
vector<int> inorder;
inorder.push_back(1);
int ptr = 0;
int nodeCount = 1;
int rounds = 0;
for (int z = 0; z < N-1; z++){
int nextnode = nextt[inorder.back()];
if (nextnode == 0) break;
trace(inorder[ptr]);
while (!goTo(nextnode)){
ptr++;
trace(inorder[ptr]);
}
nodeCount++;
inorder.push_back(nextnode);
cur = nextnode;
moves[cur] = moves[inorder[ptr]];
moves[cur].push_back(cur);
setDetails();
}
}