結果
問題 |
No.872 All Tree Path
|
ユーザー |
|
提出日時 | 2025-10-01 11:35:31 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 116 ms / 3,000 ms |
コード長 | 820 bytes |
コンパイル時間 | 3,750 ms |
コンパイル使用メモリ | 282,248 KB |
実行使用メモリ | 33,532 KB |
最終ジャッジ日時 | 2025-10-01 11:35:39 |
合計ジャッジ時間 | 6,929 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
// #pragma GCC optimize ("Ofast") // #pragma GCC optimize ("unroll-loops") // #pragma GCC target ("avx,avx2,fma") #include <bits/stdc++.h> using std::cin, std::cout, std::cerr; using ll = long long; int main() { std::ios::sync_with_stdio(false); int n; cin >> n; std::vector<std::vector<std::pair<int, int>>> e(n + 1); for(int i = 1; i < n; i ++) { int u, v, w; cin >> u >> v >> w; e[u].push_back({v, w}); e[v].push_back({u, w}); } ll ans = 0; std::vector<int> sz(n + 1); std::function<void(int, int)> dfs = [&](int x, int p) { sz[x] = 1; for(auto [i, w] : e[x]) if(i != p) { dfs(i, x); sz[x] += sz[i]; ans += ll(w) * sz[i] * (n - sz[i]) * 2; } }; dfs(1, 0); cout << ans << '\n'; }