結果
問題 | No.872 All Tree Path |
ユーザー |
|
提出日時 | 2021-01-06 08:36:36 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 115 ms / 3,000 ms |
コード長 | 1,663 bytes |
コンパイル時間 | 1,425 ms |
コンパイル使用メモリ | 137,912 KB |
最終ジャッジ日時 | 2025-01-17 09:55:16 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <cmath>#include <queue>#include <string>#include <map>#include <set>#include <stack>#include <tuple>#include <deque>#include <array>#include <numeric>#include <bitset>#include <iomanip>#include <cassert>#include <chrono>#include <random>#include <limits>#include <iterator>#include <functional>#include <sstream>#include <fstream>#include <complex>#include <cstring>#include <unordered_map>#include <unordered_set>using namespace std;using ll = long long;using P = pair<int, int>;constexpr int INF = 1001001001;constexpr int mod = 1000000007;// constexpr int mod = 998244353;template<class T>inline bool chmax(T& x, T y){if(x < y){x = y;return true;}return false;}template<class T>inline bool chmin(T& x, T y){if(x > y){x = y;return true;}return false;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int n;cin >> n;vector<vector<P>> g(n);for(int i = 1; i < n; ++i){int u, v, w;cin >> u >> v >> w;--u, --v;g[u].emplace_back(v, w);g[v].emplace_back(u, w);}vector<int> subtree_size(n, 1);ll ans = 0;auto dfs = [&](auto&& self, int from = 0, int par = -1) -> int {for(auto [to, cost] : g[from]){if(to == par) continue;subtree_size[from] += self(self, to, from);ans += (ll)cost * subtree_size[to] * (n - subtree_size[to]);}return subtree_size[from];};dfs(dfs);cout << ans * 2 << endl;return 0;}