#include<bits/stdc++.h>
using namespace std;
using LL=long long;
#define FOR(i,l,r) for(int i=(l);i<=(r);++i)
#define REP(i,n) FOR(i,0,(n)-1)
#define ssize(x) int(x.size())
template<class A,class B>auto&operator<<(ostream&o,pair<A,B>p){return o<<'('<<p.first<<", "<<p.second<<')';}
template<class T>auto operator<<(ostream&o,T x)->decltype(x.end(),o){o<<'{';int i=0;for(auto e:x)o<<(", ")+2*!i++<<e;return o<<'}';}
#ifdef DEBUG
#define debug(x...) cerr<<"["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<'\n';}(x)
#else
#define debug(...) {}
#endif
int main() {
cin.tie(0)->sync_with_stdio(0);
mt19937 rng(chrono::system_clock::now().time_since_epoch().count());
const int magic = 800;
const LL INF = (1ll << 60);
int n;
cin >> n;
vector<vector<pair<int,int>>> gg(n);
REP(i,n-1) {
int a, b, c;
cin >> a >> b >> c;
--a;
--b;
gg[a].push_back({b, c});
gg[b].push_back({a, c});
}
vector<vector<int>> g(n);
vector<int> par_cost(n);
function<void(int,int)> dfs_init = [&](int w, int p) {
for (auto [b, c] : gg[w]) {
if (b == p) continue;
dfs_init(b, w);
g[w].push_back(b);
par_cost[b] = c;
}
shuffle(g[w].begin(), g[w].end(), rng);
};
dfs_init(0, 0);
vector<vector<LL>> ans(n, vector<LL> (4));
vector<vector<vector<LL>>> dp(n, vector<vector<LL>>(2));
vector<int> sh(n);
function<void(int)> dfs = [&](int w) {
sh[w] = min(magic, ssize(g[w]) / 2 + 4);
int p = sh[w];
int s = 2 * p + 1;
dp[w] = vector<vector<LL>> (2, vector<LL> (s, -INF));
dp[w][0][0 + p] = 0;
for (auto b : g[w]) {
dfs(b);
auto dp2 = vector<vector<LL>> (2, vector<LL> (s, -INF));
REP(j,2) {
FOR(i,1,s-2) {
dp2[j][i] = max(dp2[j][i], dp[w][j][i] + ans[b][0]);
dp2[j][i] = max(dp2[j][i], dp[w][j][i - 1] + ans[b][3]);
dp2[j][i] = max(dp2[j][i], dp[w][j][i + 1] + ans[b][1]);
dp2[j][i] = max(dp2[j][i], dp[w][j ^ 1][i] + ans[b][2]);
}
}
dp[w] = dp2;
}
ans[w][0] = max(dp[w][0][0 + p], dp[w][0][1 + p] + par_cost[w]);
ans[w][1] = dp[w][0][0 + p] + par_cost[w];
ans[w][2] = dp[w][0][-1 + p] + par_cost[w];
ans[w][3] = dp[w][1][0 + p] + par_cost[w];
if (w == 0) {
ans[w][0] = dp[w][0][0 + p];
}
};
dfs(0);
cout << ans[0][0] << '\n';
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | #include<bits/stdc++.h> using namespace std; using LL=long long; #define FOR(i,l,r) for(int i=(l);i<=(r);++i) #define REP(i,n) FOR(i,0,(n)-1) #define ssize(x) int(x.size()) template<class A,class B>auto&operator<<(ostream&o,pair<A,B>p){return o<<'('<<p.first<<", "<<p.second<<')';} template<class T>auto operator<<(ostream&o,T x)->decltype(x.end(),o){o<<'{';int i=0;for(auto e:x)o<<(", ")+2*!i++<<e;return o<<'}';} #ifdef DEBUG #define debug(x...) cerr<<"["#x"]: ",[](auto...$){((cerr<<$<<"; "),...)<<'\n';}(x) #else #define debug(...) {} #endif int main() { cin.tie(0)->sync_with_stdio(0); mt19937 rng(chrono::system_clock::now().time_since_epoch().count()); const int magic = 800; const LL INF = (1ll << 60); int n; cin >> n; vector<vector<pair<int,int>>> gg(n); REP(i,n-1) { int a, b, c; cin >> a >> b >> c; --a; --b; gg[a].push_back({b, c}); gg[b].push_back({a, c}); } vector<vector<int>> g(n); vector<int> par_cost(n); function<void(int,int)> dfs_init = [&](int w, int p) { for (auto [b, c] : gg[w]) { if (b == p) continue; dfs_init(b, w); g[w].push_back(b); par_cost[b] = c; } shuffle(g[w].begin(), g[w].end(), rng); }; dfs_init(0, 0); vector<vector<LL>> ans(n, vector<LL> (4)); vector<vector<vector<LL>>> dp(n, vector<vector<LL>>(2)); vector<int> sh(n); function<void(int)> dfs = [&](int w) { sh[w] = min(magic, ssize(g[w]) / 2 + 4); int p = sh[w]; int s = 2 * p + 1; dp[w] = vector<vector<LL>> (2, vector<LL> (s, -INF)); dp[w][0][0 + p] = 0; for (auto b : g[w]) { dfs(b); auto dp2 = vector<vector<LL>> (2, vector<LL> (s, -INF)); REP(j,2) { FOR(i,1,s-2) { dp2[j][i] = max(dp2[j][i], dp[w][j][i] + ans[b][0]); dp2[j][i] = max(dp2[j][i], dp[w][j][i - 1] + ans[b][3]); dp2[j][i] = max(dp2[j][i], dp[w][j][i + 1] + ans[b][1]); dp2[j][i] = max(dp2[j][i], dp[w][j ^ 1][i] + ans[b][2]); } } dp[w] = dp2; } ans[w][0] = max(dp[w][0][0 + p], dp[w][0][1 + p] + par_cost[w]); ans[w][1] = dp[w][0][0 + p] + par_cost[w]; ans[w][2] = dp[w][0][-1 + p] + par_cost[w]; ans[w][3] = dp[w][1][0 + p] + par_cost[w]; if (w == 0) { ans[w][0] = dp[w][0][0 + p]; } }; dfs(0); cout << ans[0][0] << '\n'; } |
English