結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2019-08-30 22:30:23 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 233 ms / 3,000 ms |
コード長 | 1,426 bytes |
コンパイル時間 | 1,130 ms |
コンパイル使用メモリ | 122,360 KB |
最終ジャッジ日時 | 2025-01-07 15:59:06 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <iostream>#include <cstdio>#include <cstring>#include <vector>#include <deque>#include <queue>#include <array>#include <set>#include <map>#include <cmath>#include <algorithm>#include <numeric>#include <utility>#include <tuple>#include <functional>#include <bitset>#include <cstdint>#include <cassert>#include <random>using namespace std;using i64 = int64_t;using i32 = int32_t;template<class T, class U> void init_n(vector<T>& v, size_t n, U x){ v = vector<T>(n, x); }template<class T> void init_n(vector<T>& v, size_t n) { init_n(v, n, T()); }template<class T> void read_n(vector<T>& v, size_t n, size_t o = 0){ v = vector<T>(n+o); for (size_t i=o; i<n+o; ++i) cin >> v[i]; }template<class T> void read_n(T a[], size_t n, size_t o = 0){ for (size_t i=o; i<n+o; ++i) cin >> a[i]; }template<class T> T gabs(const T& x) { return max(x, -x); }#define abs gabsstruct E {i64 v, w;};i64 n;vector<vector<E>> g;vector<i64> ch;i64 ans;void dfs(i64 u, i64 p) {ch[u] = 1;for (auto e : g[u]) {if (e.v != p) {dfs(e.v, u);ch[u] += ch[e.v];ans += (n - ch[e.v]) * ch[e.v] * e.w;}}}int main() {cin >> n;init_n(g, n + 1);for (i64 i = 0; i < n - 1; ++i) {i64 u, v, w;cin >> u >> v >> w;g[u].push_back({v, w});g[v].push_back({u, w});}init_n(ch, n + 1);dfs(1, -1);cout << 2 * ans << endl;return 0;}