#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1000;
int depth[MAXN];
vector<int> adj[MAXN];
void dfs(int v, int pai) {
for (int viz : adj[v]) {
if (viz == pai) continue;
depth[viz] = depth[v] + 1;
dfs(viz, v);
}
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
for (int i=0; i<n; i++) {
adj[i].clear();
}
for (int i=0; i<n-1; i++) {
int a = u[i], b = v[i];
adj[a].push_back(b);
adj[b].push_back(a);
}
dfs(0, -1);
int mx = 0;
for (int i=0; i<n; i++) {
if (depth[i] > depth[mx]) {
mx = i;
}
}
dfs(mx, -1);
vector<int> labels(n);
for (int i = 0; i < n; i++) {
labels[i] = depth[i];
}
return labels;
}
int find_next_station(int s, int t, vector<int> c) {
if (s > t) {
return s-1;
} else return s+1;
}
