# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
800479 | jakobrs | 경주 (Race) (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <iostream>
#include <random>
template <typename Key, typename Value>
struct Map {
Key key_offset;
Value value_offset;
std::map<Key, Value> m;
template <typename... Args>
Map(Args &&...args)
: m(std::forward<Args>(args)...), key_offset(), value_offset() {}
void update_all_keys_by(int offset) { key_offset += offset; }
void update_all_values_by(int offset) { value_offset += offset; }
void insert(std::map<Key, Value>::value_type t) { m.insert(t); }
constexpr inline size_t size() const noexcept { return m.size(); }
};
struct Edge {
int other;
int length;
};
std::vector<std::vector<Edge>> adj;
int best;
int n, target;
void merge_into(Map<int, int> &into, Map<int, int> &from) {
if (into.size() < from.size()) {
std::swap(into, from);
}
for (auto [k_, v_] : from.m) {
auto k = k_ + from.key_offset;
auto v = v_ + from.value_offset;
auto it1 = into.m.find(target - k - into.key_offset);
if (it1 != into.m.end()) {
best = std::min(best, v + it1->second + into.value_offset);
}
}
for (auto [k_, v_] : from.m) {
auto k = k_ + from.key_offset;
auto v = v_ + from.value_offset;
auto it1 = into.m.find(k - into.key_offset);
if (it1 != into.m.end()) {
it1->second = std::min(it1->second + into.value_offset, v) -
into.value_offset;
} else {
into.insert({k - into.key_offset, v - into.value_offset});
}
}
}
Map<int, int> dfs(int node, int parent) {
Map<int, int> result;
result.insert({0, 0});
for (Edge e : adj[node]) {
if (e.other == parent) continue;
Map m = dfs(e.other, node);
m.update_all_keys_by(e.length);
m.update_all_values_by(1);
merge_into(result, m);
}
return result;
}
int best_path(int N, int K, int H[][2], int L[]) {
adj.resize(N);
best = N;
n = N;
target = K;
for (int i = 0; i < N - 1; i++) {
adj[H[i][0]].push_back({H[i][1], L[i]});
adj[H[i][1]].push_back({H[i][0], L[i]});
}
dfs(0, -1);
if (best == N)
return -1;
else
return best;
}