# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
361933 | valerikk | 기지국 (IOI20_stations) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <stations.h>
#include <vector>
std::vector<int> label(int n, int k, std::vector<int> u, std::vector<int> v) {
std::vector<std::vector<int>> e(n);
for (int i = 0; i < n - 1; i++) {
e[u[i]].push_back(v[i]);
e[v[i]].push_back(u[i]);
}
std::vector<int> in(n), out(n);
int t = 0;
function<void(int, int)> dfs = [&](int u, int p) {
in[u] = t++;
for (int v: e[u]) {
if (v != p) {
dfs(v, u);
}
}
out[u] = min(n - 1, t);
}(0, -1);
std::vector<int> res(n);
for (int i = 0; i < n; i++) res[i] = in[i] * 1000 + out[i];
}
int find_next_station(int s, int t, std::vector<int> c) {
int in_s = s / 1000, out_s = s % 1000;
if (in_s == 0) out_s++;
int in_t = t / 1000, out_t = t % 1000;
std;:vector<int> in_c, out_c;
for (int u: c) {
in_c.push_back(u / 1000);
out_c.push_back(u % 1000);
}
int p = -1;
for (size_t i = 0; i < c.size(); i++) {
if (in_c[i] <= in_s && out_c[i] >= out_s) {
p = c[i];
} else {
if (in_c[i] <= in_t && out_c[i] >= out_t) {
return c[i];
}
}
}
return p;
}