結果

問題 No.872 All Tree Path
コンテスト
ユーザー kcvlex
提出日時 2019-08-30 21:38:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,273 bytes
コンパイル時間 1,555 ms
コンパイル使用メモリ 172,748 KB
実行使用メモリ 10,112 KB
最終ジャッジ日時 2024-11-21 22:16:30
合計ジャッジ時間 3,881 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 11 RE * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

// #define DEBUGGING
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define ALL(V) (V).begin(), (V).end()
#define ALLR(V) (V).rbegin(), (V).rend()

template <typename T> using V = vector<T>;
template <typename T> using VV = V<V<T>>;
using ll = int64_t;
using ull = uint64_t;
using PLL = pair<ll, ll>;

template <typename T> const T& var_min(const T &t) { return t; }
template <typename T> const T& var_max(const T &t) { return t; }
template <typename T, typename... Tail> const T& var_min(const T &t, const Tail&... tail) { return min(t, var_min(tail...)); }
template <typename T, typename... Tail> const T& var_max(const T &t, const Tail&... tail) { return max(t, var_max(tail...)); }
template <typename T, typename... Tail> void chmin(T &t, const Tail&... tail) { t = var_min(t, tail...); }
template <typename T, typename... Tail> void chmax(T &t, const Tail&... tail) { t = var_max(t, tail...); }
template <typename T> const T& clamp(const T &t, const T &low, const T &high) { return max(low, min(high, t)); }
template <typename T> void chclamp(T &t, const T &low, const T &high) { t = clamp(t, low, high); }

namespace init__ {

struct InitIO {
    InitIO() {
        cin.tie(nullptr);
        ios_base::sync_with_stdio(false);
        cout << fixed << setprecision(30);
    }
} init_io;

}

#ifdef DEBUGGING
// #include "../debug/debug.cpp"
#include "../../debug/debug.cpp"
#else
#define DEBUG(...) 0
#define DEBUG_SEPARATOR_LINE 0
#endif

template <typename T>
T make_v(T init) { return init; }

template <typename T, typename... Tail>
auto make_v(T init, size_t s, Tail... tail) {
#define rec make_v(init, tail...)
    return V<decltype(rec)>(s, rec);
#undef rec
}

VV<PLL> edges(1e5 + 10);
ll N, ans;

ll dfs(ll cur, ll pre = -1) {
    ll ret = 1;
    for(auto &&edge : edges[cur]) {
        ll nxt, cost;
        tie(nxt, cost) = edge;
        if(nxt == pre) continue;
        ll res = dfs(nxt, cur);
        ans += 2 * res * (N - res) * cost;
        ret += res;
    }
    return ret;
}

int main() {
    cin >> N;
    for(ll i = 1; i < N; i++) {
        ll x, y, w;
        cin >> x >> y >> w;
        x--; y--;
        edges[x].emplace_back(y, w);
        edges[y].emplace_back(x, w);
    }
    dfs(0);
    cout << ans << endl;
    return 0;
}
0