#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
const int dydis = 1000;
vector<int> gr[dydis];
int enter[dydis];
int leave[dydis];
vector<int> ret (dydis);
int dbInd = 0;
void dfs(int v, int came){
enter[v] = dbInd++;
for(auto x : gr[v]){
if(x == came) continue;
dfs(x, v);
}
leave[v] = dbInd;
ret[v] = enter[v] * dydis + leave[v] - 1;
//cout << "enter " << v << " - " << enter[v] << ", o leave " << leave[v]-1 << endl;
}
void dfs1(int v, int came, int as){
ret[v] = as;
for(auto x : gr[v]){
if(x == came) continue;
dfs1(x, v, as+1);
}
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
int mxsz = 0;
for(int i = 0; i < n; i++){
gr[i].clear();
gr[i].clear();
}
dbInd = 0;
ret.resize(n);
for(int i = 0; i < n-1; i++){
gr[u[i]].push_back(v[i]);
gr[v[i]].push_back(u[i]);
}
for(int i = 0; i < n; i++) mxsz = max(mxsz, (int)gr[i].size());
if(mxsz <= 2){
int st = 0;
for(int i = 0; i < n; i++) if(gr[i].size() == 1) st = i;
dfs1(st, -1, 0);
}else{
dfs(0, -1);
}
return ret;
}
/*
int find_next_station(int s, int t, vector<int> c) {
int enterS = s / 1000;
int leaveS = s % 1000;
int enterT = t / 1000;
int leaveT = t % 1000;
// cout << "kitas einant is " << s << " i " << t << " yra:\n";
if(enterS <= enterT && leaveT <= leaveS){
// cout << enterS << " yra " << enterT << " tevas\n";
for(auto x : c){
int e = x / 1000;
int l = x % 1000;
if(e <= enterT && leaveT <= l && !(e <= enterS && leaveS <= l)){
// cout << "ret " << x << endl << endl;
return x;
}
}
}else{ // eisiu aukstyn
for(auto x : c){
int e = x / 1000;
int l = x % 1000;
if(e <= enterS && leaveS <= l){
// cout << "ret " << x << endl << endl;
return x;
}
}
}
return c[0];
}
*/
int find_next_station(int s, int t, vector<int> c) {
if(s < t) return s+1;
return s-1;
}
