#include "stations.h"
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e3 + 7;
void dfs( vector<vector<int>> adj, int v, int p, vector<int> lb, int &cnt ) {
lb[v] = cnt++;
for ( auto u: adj[v] ) {
if ( u != p ) {
dfs( adj, u, v, lb, cnt );
}
}
if ( p != -1 ) {
lb[v] *= 1000;
lb[v] += cnt - 1;
}
else {
lb[v] *= 1000;
lb[v]++;
}
}
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
vector<vector<int>> adj(n + 7);
for ( int i = 0; i < n - 1; i++ ) {
adj[ u[i] ].push_back( v[i] );
adj[ v[i] ].push_back( u[i] );
}
int mx = -1, rt = 0;
for ( int i = 0; i < n; i++ ) {
if ( (int)adj[i].size() > mx ) {
mx = adj[i].size();
rt = i;
}
}
cnt = 0;
vector<int> lb(n + 7, 0);
dfs( adj, rt, -1 , lb, cnt );
vector < int > lbs;
for ( int i = 0; i < n; i++ ) {
lbs.push_back( lb[i] );
}
return lbs;
}
int find_next_station(int s, int t, std::vector<int> c) {
int i = s / 1000;
int j = t / 1000;
if ( s == 1 ) {
for ( auto v: c ) {
int x = v / 1000, y = v % 1000;
if ( j >= x && j <= y ) {
return v;
}
}
}
int pr = -1;
for ( auto v: c ) {
int x = v / 1000, y = v % 1000;
if ( x < i ) {
pr = v;
}
if ( j >= x && j <= y ) {
return v;
}
}
assert( pr != -1 );
return pr;
}
Compilation message
stations.cpp: In function 'std::vector<int> label(int, int, std::vector<int>, std::vector<int>)':
stations.cpp:38:2: error: 'cnt' was not declared in this scope; did you mean 'int'?
38 | cnt = 0;
| ^~~
| int
