結果
問題 | 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;}