結果

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

ソースコード

diff # 

#include<bits/stdc++.h>
#define REP(i,n) for(int i=0,i##_len=(n);i<i##_len;++i)
#define rep(i,a,b) for(int i=int(a);i<int(b);++i)
#define All(x) (x).begin(),(x).end()
using namespace std;

int main(){
    long long N;cin>>N;
    typedef pair<int,long long> P;
    vector<vector<P>> graph(N);
    REP(i,N-1){
        int a,b; 
        long long c;
        cin>>a>>b>>c;
        a--;b--;
        graph[a].push_back(P(c,b));
        graph[b].push_back(P(c,a));
    }
    long long ans=0;
    vector<long long> leaf(N,0);
    auto dfs=[&](auto &&f,int i,int pre)->int{
        for(auto e:graph[i]){
            if(e.second==pre){
                continue;
            }
            leaf[i]+=f(f,e.second,i)+1;
        }
        for(auto e:graph[i]){
            if(e.second==pre) continue;
            ans+=(leaf[e.second]+1)*(N-leaf[e.second]-1)*e.first;
        }
        return leaf[i];
    };
    dfs(dfs,0,-1);
    
    cout<<2*ans<<endl;
}
0