# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
58776 | Sherazin | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "race.h"
#include <bits/stdc++.h>
#define pii pair<int, int>
#define x first
#define y second
using namespace std;
const int N = 2e5 + 5;
int n, k;
int ans = -1, centroid, mxnode, tot, cnt;
int dep[N], path[N], dp[N];
vector<vector<pii> > g(N);
bitset<N> chk;
int subtree(int u, int p) {
for(pii v : g[u]) if(!chk[v.x] && v.x != p) {
dep[u] += subtree(v.x, u);
}
return ++dep[u];
}
void centdecomp(int u, int p) {
int ret = all - dep[u];
for(pii v : g[u]) if(!chk[v.x] && v.x != p) {
centdecomp(v.x, u);
ret = max(ret, dep[v.x]);
}
if(ret < mxnode) centroid = u, mxnode = ret;
}
void dfs(int u, int p, int cost, int edge, bool fill) {
if(cost > k) return;
if(!fill) {
if(dp[k - cost] == cnt && (ans == -1 || edge + path[k - cost] < ans)) ans = edge + path[k - cost];
if(cost = k && (ans == -1 || edge < ans)) ans = edge;
} else {
if(dp[cost] < cnt) dp[cost] = cnt, path[cost] = edge;
else if(edge < path[cost]) dp[cost] = cnt, path[cost] = edge;
}
for(pii v : g[u]) if(!chk[v.x] && v.x != p) {
dfs(v.x, u, cost + v.y, edge + 1, fill);
}
}
void proc(int u) {
cnt++;
if(subtree(u, u) <= 1) return;
centroid = -1;
mxnode = dep[u];
tot = dep[u];
centdecomp(u, u, dep[u]);
for(pii v : g[centroid]) if(!chk[v.x]) {
dfs(v.x, centroid, v.y, 1, false);
dfs(v.x, centroid, v.y, 1, true);
}
chk[centroid] = 1;
for(pii v : g[centroid]) if(!chk[v.x]) {
proc(v.x);
}
}
int best_path(int N, int K, int H[][2], int L[])
{
n = N, k = K;
for(int i = 0; i < N - 1; i++) {
g[H[i][0]].emplace_back(H[i][1], L[i]);
g[H[i][1]].emplace_back(H[i][0], L[i]);
}
proc(0);
return ans;
}