# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
1200248 | timasdfadsfadsf | 경주 (Race) (IOI11_race) | C++20 | 0 ms | 0 KiB |
// time-limit: 3000
#include <bits/stdc++.h>
#include <cstdio>
#include <unordered_map>
using namespace std;
void setIO(string File_name) {
cin.tie(0)->sync_with_stdio(0);
if (File_name.size()) {
freopen((File_name + ".in").c_str(), "r", stdin);
freopen((File_name + ".out").c_str(), "w", stdout);
}
}
class CD {
private:
vector<vector<pair<long long, long long>>> tree;
vector<long long> par;
vector<long long> sub;
vector<bool> rem;
unordered_map<long long, long long> mp;
long long ans = -1;
long long k;
long long dfs(long long u, long long p) {
sub[u] = 1;
for (auto [v, w] : tree[u]) {
if (!rem[v] && v != p) {
sub[u] += dfs(v, u);
}
}
return sub[u];
}
long long dfs(long long u, long long p, long long n) {
for (auto [v, w] : tree[u]) {
if (!rem[v] && v != p && sub[v] > n / 2) {
return dfs(v, u, n);
}
}
return u;
}
void process(long long u, long long p, long long depth, long long sum, bool filling) {
long long need = k - sum;
if (need < 0) return;
if (filling) {
if (mp.find(sum) == mp.end()) {
mp[sum] = depth;
}
else {
mp[sum] = min(mp[sum], depth);
}
}
else if (mp.find(need) != mp.end()) {
if (ans == -1) {
ans = depth + mp[need];
}
else if (depth + mp[need] < ans) {
ans = depth + mp[need];
}
}
for (auto [v, w] : tree[u]) {
if (v != p && !rem[v]) {
process(v, u, depth + 1, sum + w, filling);
}
}
}
void build(long long u, long long p) {
long long n = dfs(u, p);
long long c = dfs(u, p, c);
par[c] = p;
rem[c] = true;
mp.clear();
mp[0] = 0;
for (auto [v, w] : tree[c]) {
if (!rem[v]) {
process(v, c, 1, w, 0);
process(v, c, 1, w, 1);
}
}
for (auto [v, w] : tree[c]) {
if (!rem[v]) {
build(v, u);
}
}
}
public:
CD(const vector<vector<pair<long long, long long>>>& t, long long k_) {
long long n = t.size();
sub.resize(n);
rem.resize(n);
par.resize(n);
tree = t;
k = k_;
build(0, -1);
}
long long solve() {
return ans;
}
};
struct edge {
long long u, v, w;
};
long long best_path(long long n, long long k, long long h[][2], long long* l) {
vector<edge> E(n - 1);
for (long long i = 0; i < n - 1; i++) {
long long u = h[i][0], v = h[i][1];
E[i] = {u, v, -1};
}
for (long long i = 0; i < n - 1; i++) {
long long ll = l[i];
E[i].w = ll;
}
vector<vector<pair<long long, long long>>> t(n);
for (auto e : E) {
t[e.u].push_back({e.v, e.w});
t[e.v].push_back({e.u, e.w});
}
CD g(t, k);
return g.solve();
}