#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
const int N = 1e3+5;
vector<vector<int>> adj(N);
vector<int> labels;
int timer = 0;
vector<int> vis(N, 0);
void dfs(int u){
labels[u] = timer++;
vis[u] = true;
for(auto &v : adj[u]){
if(vis[v]) continue;
dfs(v);
}
}
vector<int> label(int n, int k, vector<int> u, vector<int> v){
labels.resize(n);
for(int i = 0; i < n-1; i++){
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
}
int rt = 0;
for(int i = 0; i < n; i++){
if((int)adj[i].size() == 1) { rt = i; break; }
}
dfs(rt);
return labels;
}
int find_next_station(int s, int t, vector<int> c){
if((int)c.size() == 1) return c[0];
if(t > s) return c[1];
else return c[0];
}
