結果
問題 | No.872 All Tree Path |
ユーザー |
|
提出日時 | 2019-08-30 21:41:20 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 118 ms / 3,000 ms |
コード長 | 1,205 bytes |
コンパイル時間 | 1,681 ms |
コンパイル使用メモリ | 171,912 KB |
実行使用メモリ | 23,764 KB |
最終ジャッジ日時 | 2024-11-21 22:22:01 |
合計ジャッジ時間 | 3,620 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#define _USE_MATH_DEFINES #include "bits/stdc++.h" using namespace std; #define FOR(i,j,k) for(int (i)=(j);(i)<(int)(k);++(i)) #define rep(i,j) FOR(i,0,j) #define each(x,y) for(auto &(x):(y)) #define mp make_pair #define MT make_tuple #define all(x) (x).begin(),(x).end() #define debug(x) cout<<#x<<": "<<(x)<<endl #define smax(x,y) (x)=max((x),(y)) #define smin(x,y) (x)=min((x),(y)) #define MEM(x,y) memset((x),(y),sizeof (x)) #define sz(x) (int)(x).size() #define RT return using ll = long long; using pii = pair<int, int>; using vi = vector<int>; using vll = vector<ll>; vector<pii> G[200005]; ll ans; int N; int dfs(int u, int p) { int res = 1; each(e, G[u]) { if (e.first != p) { int num = dfs(e.first, u); ans += (ll)num * (N - num) * e.second * 2; res += num; } } return res; } void solve() { cin >> N; rep(i, N - 1) { int u, v, w; cin >> u >> v >> w; --u; --v; G[u].emplace_back(v, w); G[v].emplace_back(u, w); } dfs(0, -1); cout << ans << endl; } int main() { ios::sync_with_stdio(false); cin.tie(0); cout << fixed << setprecision(15); solve(); return 0; }