結果

問題 No.872 All Tree Path
コンテスト
ユーザー Hoi_koro
提出日時 2019-08-30 22:00:46
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 150 ms / 3,000 ms
コード長 1,360 bytes
コンパイル時間 2,118 ms
コンパイル使用メモリ 175,312 KB
実行使用メモリ 29,952 KB
最終ジャッジ日時 2024-11-21 23:22:37
合計ジャッジ時間 4,072 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>

#if MYDEBUG
#include "lib/cp_debug.hpp"
#else
#define DBG(...) ;
#endif
#if __cplusplus <= 201402L
template <typename T>
T gcd(T a, T b) { return ((a % b == 0) ? b : gcd(b, a % b)); }
template <typename T>
T lcm(T a, T b) { return a / gcd(a, b) * b; }
#endif
using LL = long long;
constexpr LL LINF = 334ll << 53;
constexpr int INF = 15 << 26;
constexpr LL MOD = 1E9 + 7;

namespace Problem {
using namespace std;

class Solver {
 public:
  int n;
  vector<vector<pair<int, LL>>> t;
  vector<LL> cnt;
  Solver(LL n) : n(n), t(n), cnt(n){};

  LL dfs(int c, int p = -1) {
    DBG(c, p)
    LL ret = 0;
    LL subcnt = 1;
    LL len = 0;
    for (auto e : t[c]) {
      if (e.first != p) {
        ret += dfs(e.first, c);
        subcnt += cnt[e.first];
      } else {
        len = e.second;
      }
    }
    cnt[c] = subcnt;
    return ret + 2 * len * subcnt * (n - subcnt);
  }

  void solve() {
    for (int i = 0; i < n - 1; ++i) {
      int u, v, w;
      cin >> u >> v >> w;
      t[--u].emplace_back(--v, w);
      t[v].emplace_back(u, w);
    }
    cout << dfs(0) << endl;
  }
};

}  // namespace Problem

int main() {
  std::cin.tie(0);
  std::ios_base::sync_with_stdio(false);
  // std::cout << std::fixed << std::setprecision(12);
  long long n = 0;
  std::cin >> n;

  Problem::Solver sol(n);
  sol.solve();
  return 0;
}
0