#include <bits/stdc++.h>
#include "speedrun.h"
using namespace std;
void dfsHint(int curr, int par, vector<vector<int>> &adjHint, vector<int> &path)
{
path.push_back(curr);
for(int i = 0; i < 10; i++) // set in big-endian
{
if(par & (1 << i)) setHint(curr, i + 1, 1);
else setHint(curr, i + 1, 0);
}
for(int next : adjHint[curr])
{
if(next == par) continue;
dfsHint(next, curr, adjHint, path);
}
}
void assignHints(int subtask, int N, int A[], int B[])
{
setHintLen(20);
vector<vector<int>> adjHint(N + 1);
for(int i = 1; i < N; i++)
{
adjHint[A[i]].push_back(B[i]);
adjHint[B[i]].push_back(A[i]);
}
for(int i = 1; i <= N; i++) sort(adjHint[i].begin(), adjHint[i].end());
vector<int> path;
dfsHint(1, 0, adjHint, path);
// for(auto i : path) cout << i << " ";
// cout << "\n";
for(int i = 0; i < path.size() - 1; i++)
{
int child = path[i + 1];
for(int j = 10; j < 20; j++) // set in big-endian
{
if(child & (1 << (j - 10))) setHint(path[i], j + 1, 1);
else setHint(path[i], j + 1, 0);
}
}
}
int dfs(int curr, int toTry) // returns next value to try
{
int par = 0, child = 0;
for(int i = 0; i < 10; i++) par += getHint(i + 1) * (1 << i);
for(int i = 10; i < 20; i++) child += getHint(i + 1) * (1 << (i - 10));
if(toTry == 0) toTry = child;
if(toTry == 0) return 0;
while(goTo(toTry))
{
toTry = dfs(toTry, 0);
}
goTo(par);
return toTry;
}
void speedrun(int subtask, int N, int start)
{
if(N == 1) return;
int l = getLength();
while(start != 1)
{
int par = 0;
for(int i = 0; i < 10; i++) par += getHint(i + 1) * (1 << i);
goTo(par);
start = par;
}
dfs(start, 0);
}
Compilation message
speedrun.cpp: In function 'void assignHints(int, int, int*, int*)':
speedrun.cpp:35:19: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
35 | for(int i = 0; i < path.size() - 1; i++)
| ~~^~~~~~~~~~~~~~~~~
speedrun.cpp: In function 'void speedrun(int, int, int)':
speedrun.cpp:66:6: warning: unused variable 'l' [-Wunused-variable]
66 | int l = getLength();
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
187 ms |
796 KB |
Invalid node index to goTo |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
185 ms |
740 KB |
Invalid node index to goTo |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
180 ms |
756 KB |
Invalid node index to goTo |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
128 ms |
744 KB |
Invalid node index to goTo |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
133 ms |
768 KB |
Invalid node index to goTo |
2 |
Halted |
0 ms |
0 KB |
- |