# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
416279 | DEQK | 경주 (Race) (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "race.h"
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 200200;
ll ans = -1;
ll d[N], K;
ll len[N];
vector<pair<int, int>> g[N];
void calc(int u,int p,ll x = 0) {
len[u] = x;
for(auto to : g[u]) if(to.first != p) {
d[to.first] = d[u] + 1;
calc(to.first, u, x + to.second);
}
}
map<ll, ll> dfs(int u,int p) {
map<ll, ll> own;
own[len[u]] = d[u];
for(auto to : g[u]) if(to.first != p) {
auto x = dfs(to.first, u);
if(x.size() > own.size()) {
own.swap(x);
}
for(auto it : x) {
if(own.count(K - it.first + 2 * len[u])) {
if(ans == -1) {
ans = own[K - it.first + 2 * len[u]] + it.second - 2 * d[u];
} else {
ans = min(ans, own[K - it.first + 2 * len[u]] + it.second - 2 * d[u]);
}
}
if(own.count(it.first)) {
own[it.first] = min(own[it.first], it.second);
} else {
own[it.first] = it.second;
}
}
}
return own;
}
int best_path(int n, int k, int h[][2], ll l[]) {
K = k;
for(int i = 0; i < n - 1; i++) {
ll u = h[i][0],v = h[i][1], w = l[i];
g[u].push_back({v, w});
g[v].push_back({u, w});
}
calc(0, -1);
dfs(0, -1);
return ans;
}
/*int main() {
int n, k; cin >> n >> k;
int h[n - 1][2];
for(int i = 0; i < n - 1; ++i) {
cin >> h[i][0] >> h[i][1];
}
for(int i = 0; i < n - 1; ++i) {
}
}*/