結果
問題 | No.872 All Tree Path |
ユーザー |
![]() |
提出日時 | 2023-05-03 15:44:19 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 173 ms / 3,000 ms |
コード長 | 1,550 bytes |
コンパイル時間 | 2,132 ms |
コンパイル使用メモリ | 199,904 KB |
最終ジャッジ日時 | 2025-02-12 16:38:57 |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h>//#include <atcoder/all>using namespace std;//using namespace atcoder;//using mint = modint1000000007;//const int mod = 1000000007;//using mint = modint998244353;//const int mod = 998244353;//const int INF = 1e9;//const long long LINF = 1e18;#define rep(i, n) for (int i = 0; i < (n); ++i)#define rep2(i,l,r)for(int i=(l);i<(r);++i)#define rrep(i, n) for (int i = (n-1); i >= 0; --i)#define rrep2(i,l,r)for(int i=(r-1);i>=(l);--i)#define all(x) (x).begin(),(x).end()#define allR(x) (x).rbegin(),(x).rend()#define endl "\n"#define P pair<long long,int>template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }struct Edge {int to;long long d;Edge(int _to, long long _d) :to(_to), d(_d) {}};int main() {ios::sync_with_stdio(false);cin.tie(nullptr);int n; cin >> n;vector<vector<Edge>>to(n);rep(i, n - 1) {int a, b, c; cin >> a >> b >> c;a--, b--;to[a].emplace_back(b, c);to[b].emplace_back(a, c);}auto dfs = [&](auto self, int v, int p = -1)->P {P ret = { 0,1 };for (auto nv : to[v]) {if (p == nv.to)continue;auto get = self(self, nv.to, v);long long r = get.second;long long l = n - r;ret.first += r * l * nv.d;ret.first += get.first;ret.second += get.second;}return ret;};auto ans = dfs(dfs, 0);cout << ans.first * 2 << endl;return 0;}