# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
104356 | nvmdava | Race (IOI11_race) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#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);
}