# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
419834 | dxz05 | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#pragma GCC optimize("Ofast")
#include "race.h"
#include <bits/stdc++.h>
using namespace std;
const int MAXN = 2e5 + 3e2;
vector<pair<int, int>> g[MAXN];
int k;
int D[MAXN], P[MAXN];
int dfs(int v, int p, int dist, int cnt, vector<pair<int,int> > &dists) {
dists.push_back({dist,cnt});
for (auto [x,y]:g[v]) {
if (x!=p&&D[x]!=-1) {
dfs(x,v, dist +y,cnt+1, dists);
}
}
}
int get_centroid(int v, int p, int n, int ¢roid){
int size = 1;
for (auto edge : g[v]){
int u = edge.first, w = edge.second;
if (D[u] == -1 && u != p){
size += get_centroid(u, v, n, centroid);
}
}
if (size >= (n + 1) / 2 && centroid == -1) centroid = v;
return size;
}
int ans = 1e9;
vector<int> d(1000001,-1);
void build_centroid_tree(int v, int p, int dep, int dist, int n){
int centroid = -1;
get_centroid(v, -1, n, centroid);
if (centroid == -1) centroid = v;
D[centroid] = dep;
P[centroid] = p;
vector<int> changed;
for (auto edge : g[centroid]){
int u = edge.first, w = edge.second;
if (u == p || D[u] == -1) continue;
vector<int> dists;
dfs(u, v, w, 1, dists);
for (auto &now : dists){
int x = now.first, cnt = now.second;
if (k >= x && d[k - x] != -1) {
ans = min(ans, d[k - x] + cnt);
}
}
for (auto &now : dists){
int x = now.first, cnt = now.second;
if (x >= d.size()) continue;
if (d[x] == -1 || d[x] > cnt) {
changed.push_back(x);
d[x] = cnt;
}
}
}
for (auto x:changed) d[x] = -1;
for (auto edge : g[centroid]){
int u = edge.first, w = edge.second;
build_centroid_tree(u, centroid, dep + 1, w, (n + 1) / 2);
}
return vec;
}
int best_path(int N, int K, int H[][2], int L[]){
k = K;
d[0] = 0;
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]);
}
build_centroid_tree(1, -1, 1, 0, N);
if (ans == 1e9) ans = -1;
return ans;
}
/*
4 3
0 1 1
1 2 2
1 3 4
2
3 3
0 1 1
1 2 1
-1
11 12
0 1 3
0 2 4
2 3 5
3 4 4
4 5 6
0 6 3
6 7 2
6 8 5
8 9 6
8 10 7
2
*/