結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2020-01-18 18:52:47 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 669 ms / 3,000 ms |
コード長 | 1,236 bytes |
コンパイル時間 | 1,369 ms |
コンパイル使用メモリ | 108,120 KB |
最終ジャッジ日時 | 2025-01-08 20:09:25 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <algorithm>#include <cmath>#include <complex>#include <iostream>#include <map>#include <queue>#include <set>#include <vector>using namespace std;template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }using ll = long long;const long long MOD = 1e9+7;struct Edge {int u, v;ll w;};int max_v = 200000;vector<vector<int>> graph(max_v);vector<int> cnt_child(max_v);vector<bool> done(max_v);int dfs(int v) {if (done.at(v)) return 0;done.at(v) = true;int res = 1;for (int to: graph.at(v)) res += dfs(to);return cnt_child.at(v) = res;}int main() {int n;cin >> n;vector<Edge> edges(n - 1);for (int i = 0; i < n - 1; i++) {int u, v;ll w;cin >> u >> v >> w;u--;v--;graph.at(u).push_back(v);graph.at(v).push_back(u);edges.at(i) = Edge{u, v, w};}dfs(0);ll sum = 0;for (Edge e: edges) {ll m = min(cnt_child.at(e.u), cnt_child.at(e.v));sum += m * (n - m) * e.w;}cout << sum * 2 << endl;}