| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 104356 | nvmdava | Race (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "race.h"
#include <bits/stdc++.h>
#define N 200005
#define pb push_back
#define pii pair<int, int>
bool blocked[N];
int n, s;
vector<pii> adj[N];
int sz[N];
int sumsz;
int len[5 * N];
void get_size(int v, int p){
sz[v] = 1;
len[v] = 0x3f3f3f3f;
for(auto& c : adj[v]){
if(c.ff == p || blocked[c.ff])
continue;
get_size(c.ff, v);
sz[v] += sz[c.ff];
}
}
int get_centroid(int v, int p){
for(auto& c : adj[v]){
if(c.ff == p || blocked[c.ff])
continue;
if(sz[c.ff] >= sumsz / 2)
return get_centroid(c.ff, v);
}
return v;
}
int ans = N;
queue<pii> q;
void dfs(int v, int p, int l, int cnt){
ans = min(ans, len[s - l] + cnt);
q.push({l, cnt});
cnt++;
for(auto& c : adj[v]){
if(c.ff == p || blocked[c])
continue;
dfs(c.ff, v, l + c.ss, cnt);
}
if(p == -1){
while(!q.empty()){
len[q.top().ff] = min(len[q.top().ff], q.top().ss);
q.pop();
}
}
}
void decomp(int v){
get_size(v, -1);
v = get_centroid(v, -1);
blocked[v] = 1;
for(auto &c : adj[v]){
if(blocked[c.ff])
continue;
dfs(c.ff, -1, 0, 0);
}
for(auto &c : adj[v]){
if(blocked[c.ff])
continue;
decomp(c.ff);
}
}
int best_path(int n, int s, int h[][2], int l[]){
::n = n;
::s = s;
for(int i = 0; i < n - 1; i++){
adj[h[i][0]].pb({h[i][1], l[i]});
adj[h[i][1]].pb({h[i][0], l[i]});
}
decomp(1);
return (ans == N ? -1 : ans);
}