結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2019-08-30 21:51:31 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 162 ms / 3,000 ms |
コード長 | 850 bytes |
コンパイル時間 | 1,576 ms |
コンパイル使用メモリ | 176,388 KB |
実行使用メモリ | 28,304 KB |
最終ジャッジ日時 | 2024-11-21 22:42:08 |
合計ジャッジ時間 | 3,764 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h> int ri() { int n; assert(scanf("%d", &n) == 1); return n; } int n; std::vector<std::vector<std::pair<int, int> > > hen; std::vector<int> size; std::vector<int> parent; int dfs(int i, int par) { parent[i] = par; int cur = 1; for (auto &j : hen[i]) { if (j.first == par) continue; cur += dfs(j.first, i); } return size[i] = cur; } int main () { n = ri(); hen.resize(n); for (int i = 0; i + 1 < n; i++) { int a = ri() - 1; int b = ri() - 1; int c = ri(); hen[a].push_back({b, c}); hen[b].push_back({a, c}); } size.resize(n); parent.resize(n); dfs(0, -1); int64_t res = 0; for (int i = 0; i < n; i++) { for (auto &j : hen[i]) { if (parent[j.first] == i) { res += j.second * (int64_t) size[j.first] * (n - size[j.first]); } } } std::cout << res * 2 << std::endl; return 0; }