#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
using vi = vector<int>;
vi labels;
vector<vi>adjL;
int dfs(int pos, int cnt, int prev=-1) {
for(int adj: adjL[pos]) {
if(adj == prev) continue;
cnt = dfs(adj, cnt, pos);
}
labels[pos] = cnt;
return cnt+1;
}
vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
labels.resize(n);
adjL.assign(n, vi());
for(int i=0; i<n-1; ++i) {
adjL[u[i]].push_back(v[i]);
adjL[v[i]].push_back(u[i]);
}
dfs(0, 0);
return labels;
}
int find_next_station(int s, int t, std::vector<int> c) {
int n = (int)c.size();
if(c.back() > s) --n;
for(int i=0; i<n; ++i) {
int nxt;
if(i == n-1) nxt = s;
else nxt = c[i+1];
if(t >= c[i] && t < nxt) return c[i];
}
return c.back();
}
