# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
306092 | SorahISA | 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;
using pii = pair<int, int>;
template<typename T>
using Prior = priority_queue<T>;
template<typename T>
using prior = priority_queue<T, vector<T>, greater<T>>;
#define X first
#define Y second
#define ALL(x) (x).begin(), (x).end()
#define eb emplace_back
#define pb push_back
const int maxn = 1000 + 5;
const int INF = 1E9;
int lstvis = 0;
vector<int> adj[maxn], lab(n, INF);
void dfs(int now, int lst) {
for (auto x : adj[now]) {
if (x != lst) lab[x] = lab[now] + 1, lstvis = x, dfs(x, now);
}
}
vector<int> label(int n, int k, vector<int> u, vector<int> v) {
for (int i = 0; i < n-1; ++i) {
adj[u[i]].eb(v[i]), adj[v[i]].eb(u[i]);
}
dfs(0, -1), lab[lstvis] = 0, dfs(lstvis, -1);
return lab;
}
int find_next_station(int s, int t, vector<int> c) {
return s > t ? c.front() : c.back();
}