#include <bits/stdc++.h>
#include "stations.h"
//#include "stub.cpp"
using namespace std;
#define pb push_back
vector <int> adj[1001];
int in[1001], out[1001];
int cnt = -1;
void dfs (int node, int last)
{
cnt++;
in[node] = cnt;
for (auto next : adj[node])
{
if (next==last)
{
continue;
}
dfs(next, node);
}
out[node] = cnt;
}
bool check(int lab1, int lab2)
{
int cur_in = lab1/1000;
int cur_out = lab1%1000;
int des_in = lab2/1000;
int des_out = lab2%1000;
if (cur_in < des_in and des_in <= cur_out)
{
return true;
}
else
{
return false;
}
}
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v)
{
memset(in, 0, sizeof(in));
memset(out,0,sizeof(out));
vector <int> labels;
for (int i = 0; i < n-1; i++)
{
adj[u[i]].pb(v[i]);
adj[v[i]].pb(u[i]);
}
dfs(0, -1);
for (int i=0; i<n; i++)
{
labels.pb(in[i]*1000 + out[i]);
}
return labels;
}
int find_next_station(int s, int t, std::vector<int> c)
{
if (check(s, t))
{
for (int child : c)
{
if (child==t)
{
return child;
}
if (check(child, s))
{
continue;
}
if (check(child, t))
{
return child;
}
}
}
else
{
for (int child : c)
{
if (child==t)
{
return child;
}
if (check(child, s))
{
return child;
}
}
}
return c[0];
}
/*
1
5 10
0 1
1 2
1 3
2 4
2
2 0 1
1 3 3
*/
Compilation message
stations.cpp: In function 'bool check(int, int)':
stations.cpp:32:9: warning: unused variable 'des_out' [-Wunused-variable]
32 | int des_out = lab2%1000;
| ^~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
2469 ms |
2097156 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Execution timed out |
3055 ms |
308 KB |
Time limit exceeded |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
1564 ms |
2097156 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
922 ms |
584 KB |
Output is correct |
2 |
Runtime error |
1180 ms |
2097156 KB |
Execution killed with signal 9 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
2822 ms |
2097156 KB |
Execution killed with signal 9 |
2 |
Halted |
0 ms |
0 KB |
- |