結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2020-11-14 11:22:05 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 326 ms / 3,000 ms |
コード長 | 1,024 bytes |
コンパイル時間 | 2,054 ms |
コンパイル使用メモリ | 179,504 KB |
実行使用メモリ | 42,244 KB |
最終ジャッジ日時 | 2024-07-22 22:46:16 |
合計ジャッジ時間 | 5,968 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h>using namespace std;void dfs(vector<int> &dp, vector<vector<pair<int, int>>> &c, int v = 0){dp[v] = 1;for (auto P : c[v]){int w = P.second;dfs(dp, c, w);dp[v] += dp[w];}}int main(){int N;cin >> N;vector<vector<pair<int, int>>> E(N);for (int i = 0; i < N - 1; i++){int u, v, w;cin >> u >> v >> w;u--;v--;E[u].push_back(make_pair(w, v));E[v].push_back(make_pair(w, u));}vector<int> p(N, -1);vector<vector<pair<int, int>>> c(N);queue<int> Q;Q.push(0);while (!Q.empty()){int v = Q.front();Q.pop();for (auto P : E[v]){int w = P.second;if (w != p[v]){p[w] = v;c[v].push_back(P);Q.push(w);}}}vector<int> dp(N);dfs(dp, c);long long ans = 0;for (int i = 0; i < N; i++){for (auto P : c[i]){int d = P.first;int s = P.second;ans += (long long) d * dp[s] * (N - dp[s]);}}ans *= 2;cout << ans << endl;}