# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1049986 | otarius | 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 <bits/stdc++.h>
#include <bits/extc++.h>
using namespace __gnu_pbds;
using namespace std;
#pragma GCC optimize("Ofast")
#pragma GCC optimize ("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define ff first
#define sc second
#define pb push_back
#define ll long long
#define pll pair<ll, ll>
#define pii pair<int, int>
#define ull unsigned long long
#define all(x) (x).begin(),(x).end()
// #define int long long
// #define int unsigned long long
#define ordered_set(T) tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>
// #define ordered_multiset(T) tree<T, null_type, less_equal<T>, rb_tree_tag, tree_order_statistics_node_update>
void open_file(string filename) {
freopen((filename + ".in").c_str(), "r", stdin);
freopen((filename + ".out").c_str(), "w", stdout);
}
// const ll mod = 1e9 + 7;
// const ll mod = 998244353;
const ll inf = 1e9;
const ll biginf = 1e18;
const int maxN = 5 * 1e5 + 5;
map<int, int> dp[maxN];
int dep[maxN], sum[maxN], ans = inf;
void dfs(int v, int par, int len, int val) {
dp[v][sum[v]] = dep[v];
dep[v] = len; sum[v] = val;
for (pii u : g[v]) {
if (u.ff == par) continue;
dfs(u.ff, v, len + 1, val + u.sc);
for (pii u : g[v]) {
if (u.ff == par) continue;
if (dp[u.ff].size() > dp[v].size())
swap(dp[u.ff], dp[v]);
for (auto x : dp[u.ff]) {
if (dp[v].find(k + 2 * sum[v] - x.ff) != dp[v].end()) {
ans = min(ans, dp[v][k + 2 * sum[v] - x.ff] + x.sc - 2 * dep[v]);
}
} for (auto x : dp[u.ff]) {
if (dp[v].find(x.ff) == dp[v].end())
dp[v][x.ff] = x.sc;
else dp[v][x.ff] = min(dp[v][x.ff], x.sc);
}
}
}
}
int best_path(int _n, int _k, int h[][2], int l[]) {
for (int i = 0; i < n - 1; i++) {
g[h[i][0]].pb({h[i][1], l[i]});
g[h[i][1]].pb({h[i][0], l[i]});
} dfs(0, -1, 0, 0);
return ans;
}