# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1141903 | AtabayRajabli | Race (IOI11_race) | C++17 | 0 ms | 0 KiB |
#include "race.h"
#include <bits/stdc++.h>
#define all(v) v.begin(), v.end()
using namespace std;
const int sz = 2e5 + 5, inf = 1e9 + 7;
int n, m, k, ans, w[sz], len[sz], d[sz], used[sz], par[sz];
vector<array<int, 2>> g[sz];
vector<int> cur, tot;
void dfs(int v, int p)
{
w[v] = 1;
for(int i : g[v])
{
if(i == p) continue;
dfs(i, v);
w[v] += w[i];
}
}
int fnd(int v, int p = 0)
{
for(int i : g[v])
{
if(i == p || used[i]) continue;
if(w[i] > w[v] / 2)
{
w[v] -= w[i];
w[i] += w[v];
return fnd(i, v);
}
}
return v;
}
void dist(int v, int p, int cst)
{
len[v] = len[p] + 1;
d[v] = d[p] + cst;
cur.push_back(v);
for(auto i : g[v])
{
if(i[0] == p || used[i[0]]) continue;
dist(i[0], v, x[1]);
}
}
void build(int v)
{
int c = fnd(v);
used[c] = 1;
mp[0] = 0;
for(auto x : g[c])
{
int i = x[0];
if(used[i]) continue;
cur.clear();
dist(i, v);
for(int j : cur)
{
if(mp.find(k - d[j]) == mp.end()) continue;
ans = min(ans, mp[k - d[j]] + len[j]);
}
for(int j : cur)
{
mp[d[j]] = min(mp[d[j]], len[j]);
}
}
mp.clear();
for(auto x : g[c])
{
if(used[x[0]]) continue;
build(x[1]);
}
}
int best_path(int N, int K, int H[][2], int L[])
{
n = N, k = K;
for(int i = 0; i < n - 1; i++)
{
int u = h[i][0], v = h[i][1];
cin >> u >> v;
g[u].push_back({v, l[i]});
g[v].push_back({u, l[i]});
}
dfs(1, 1);
build(1);
cout << ans;
}