# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
424197 | OttoTheDino | Stations (IOI20_stations) | C++17 | 0 ms | 0 KiB |
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 rep(i,s,e) for (int i = s; i <= e; ++i)
#define pb push_back
typedef vector<int> vi;
const int mxn = 2e4;
int id, res[mxn];
vi neibs[mxn];
void dfs (int u, int prev) {
for (int v : neibs[u]) {
if (v==prev) continue;
dfs (v, u);
}
res[u]=id++;
}
vi label (int n, int k, vi u, vi v) {
rep (i,0,n-2) {
neibs[u[i]].pb(v[i]);
neibs[v[i]].pb(u[i]);
}
return -1;
dfs (0,-1);
vi L(n);
rep (i,0,n-1) L[i] = res[i];
return L;
}
int find_next_station (int s, int t, vi c) {
sort(c.begin(), c.end());
for (int v : c) if (v>=t) return v;
return c.back();
}