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 <bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i=0;i<n;++i)
#define pb push_back
#define all(x) x.begin(), x.end()
using ll = long long;
const int N=1e3;
int l[N], r[N];
vector<int> adj[N];
int nxt=0;
void dfs(int u, int p) {
	l[u]=r[u]=nxt++;
	for(auto&v:adj[u]) {
		if (v==p) continue;
		dfs(v,u);
		r[u]=max(r[u],r[v]);
	}
}
vector<int> label(int n, int k, vector<int>u, vector<int> v) {
	forn(i,n-1) {
		adj[u[i]].pb(v[i]);
		adj[v[i]].pb(u[i]);
	}
	dfs(0,-1);
	vector<int> ans(n);
	forn(i,n) ans[i]=l[i]+N*r[i];
	return ans;
}
int find_next_station(int s, int t, vector<int> adj) {
	int l=s%N, r=s/N;
	int lx=t%N, rx=t/N;
	if (l<=lx && rx<=r) {
		for(auto&x:adj) {
			int lq=x%N, rq=x/N;
			if (lq<l) continue;
			if (lq<=lx && rx<=rq) return x;
		}
	} else {
		for(auto&x:adj) {
			int lq=x%N, rq=x/N;
			if (lq<l) return x;
		}
	}
	//assert(0);
}
Compilation message (stderr)
stations.cpp: In function 'int find_next_station(int, int, std::vector<int>)':
stations.cpp:44:16: warning: unused variable 'rq' [-Wunused-variable]
   44 |    int lq=x%N, rq=x/N;
      |                ^~
stations.cpp:49:1: warning: control reaches end of non-void function [-Wreturn-type]
   49 | }
      | ^| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... | 
| # | Verdict | Execution time | Memory | Grader output | 
|---|
| Fetching results... |