| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 943353 | THXuan | Race (IOI11_race) | C++14 | 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 <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <queue>
#include <set>
#include <map>
#define INF 1e9
using namespace std;
typedef long long ll;
int n, k;
ll ans = INF;
vector<pair<int, int>>adj[200005];
bool visited[200005];
ll dp[200005], cnt[1000005];
vector<ll>v;
int get_size(int s, int e) {
dp[s] = 1;
for (auto u : adj[s]) {
if (u.first == e || visited[u.first]) continue;
get_size(u.first, s);
dp[s] += dp[u.first];
}
return dp[s];
}
int get_centroid(int s, int e, int n) {
for (auto u : adj[s]) {
if (u.first == e || visited[u.first]) continue;
if (dp[u.first] * 2 > n)return get_centroid(u.first, s, n);
}
return s;
}
void dfs(int s, int e, bool flag, int sum, ll dep = 1) {
if (sum > k) return;
if (!flag) {
ans = min(ans, dep + cnt[k - sum]);
}
else {
cnt[sum] = min(cnt[sum], dep);
v.push_back(sum);
}
for (auto u : adj[s]) {
if (u.first == e || visited[u.first]) continue;
dfs(u.first, s, flag, sum + u.second, dep + 1);
}
}
void centroid_decomposition(int node) {
int centroid = get_centroid(node, 0, get_size(node, 0));
visited[centroid] = true;
for (auto u : adj[centroid]) {
if (!visited[u.first]) {
dfs(u.first, centroid, false, u.second);
dfs(u.first, centroid, true, u.second);
}
}
for (auto u : v) cnt[u] = INF;
v.clear();
for (auto u : adj[centroid]) {
if (!visited[u.first])centroid_decomposition(u.first);
}
}
int best_path(int a, int b, int h[][2], int l[]) {
n = a; k = b;
for (int i = 1; i <= n - 1; i++) {
int u = h[i][0], v = h[i][1], w = l[i];
adj[u].push_back({ v, w });
adj[v].push_back({ u, w });
}
for (int i = 1; i <= 1000005; i++)cnt[i] = INF;
centroid_decomposition(1);
if (ans == INF) return -1;
else return ans;
}
int h[200005][2], l[200005];
int main()
{
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> n >> k;
for (int i = 1; i <= n - 1; i++) {
cin >> h[i][0] >> h[i][1];
cin >> l[i];
}
cout << best_path(n, k, h, l) << "\n";
}