# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
308625 | szekelymilan | Stations (IOI20_stations) | C++14 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "stations.h"
#include <iostream>
#include <vector>
const int P = 2000;
std::vector<std::vector<int>> G;
std::vector<int> labels;
int time;
void DFS(int node = 0) {
labels[node] = (time++) * P;
for (int v : G[node])
if (labels[v] == -1)
DFS(v);
labels[node] += time++;
}
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
G.assign(n, std::vector<int>());
labels.assign(n, -1);
time = 0;
for (int i = 0; i < n - 1; i++) {
G[u[i]].push_back(v[i]);
G[v[i]].push_back(u[i]);
}
DFS();
return labels;
}
inline bool inside(int a, int b) {
return a / P <= b / P && b % P <= a % P;
}
int find_next_station(int s, int t, std::vector<int> c) {
int parent = -1;
for (int node : c)
if (inside(node, s))
parent = node;
for (int node : c)
if (node != parent && inside(node, t))
return node;
return parent;
}