結果

問題 No.872 All Tree Path
ユーザー ttttanttttan
提出日時 2019-08-31 17:59:16
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 350 ms / 3,000 ms
コード長 2,781 bytes
コンパイル時間 1,682 ms
コンパイル使用メモリ 168,560 KB
実行使用メモリ 37,336 KB
最終ジャッジ日時 2024-05-03 20:45:06
合計ジャッジ時間 6,495 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 349 ms
24,928 KB
testcase_01 AC 347 ms
23,160 KB
testcase_02 AC 350 ms
23,984 KB
testcase_03 AC 208 ms
37,336 KB
testcase_04 AC 5 ms
8,892 KB
testcase_05 AC 338 ms
24,192 KB
testcase_06 AC 339 ms
24,096 KB
testcase_07 AC 333 ms
23,088 KB
testcase_08 AC 25 ms
10,124 KB
testcase_09 AC 26 ms
10,816 KB
testcase_10 AC 25 ms
10,920 KB
testcase_11 AC 25 ms
9,612 KB
testcase_12 AC 25 ms
9,796 KB
testcase_13 AC 4 ms
9,200 KB
testcase_14 AC 4 ms
9,196 KB
testcase_15 AC 4 ms
8,628 KB
testcase_16 AC 4 ms
9,672 KB
testcase_17 AC 4 ms
8,964 KB
testcase_18 AC 4 ms
8,716 KB
testcase_19 AC 5 ms
9,296 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
//ios::sync_with_stdio(false);
//cin.tie(0);
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<pii,int> ppii;
typedef pair<int,pii> pipi;
typedef pair<ll,ll> pll;
typedef pair<pll,ll> ppll;
typedef pair<ll,pll> plpl;
typedef tuple<ll,ll,ll> tl;
ll mod=1000000007;
ll mod2=998244353;
ll inf=1000000000000000000;
double pi=2*acos(0);
#define rep(i,m,n) for(ll i=m;i<n;i++)
#define rrep(i,n,m) for(ll i=n;i>=m;i--)
int dh[4]={1,-1,0,0};
int dw[4]={0,0,1,-1};
int ddh[8]={-1,-1,-1,0,0,1,1,1};
int ddw[8]={-1,0,1,-1,1,-1,0,1};
ll lmax(ll a,ll b){
    if(a<b)return b;
    else return a;
}
ll lmin(ll a,ll b){
    if(a<b)return a;
    else return b;
}
ll gcd(ll a,ll b){
    if(a<b)swap(a,b);
    if(a%b==0)return b;
    return gcd(b,a%b);
}
ll Pow(ll n,ll k){
    ll ret=1;
    ll now=n;
    while(k>0){
        if(k&1)ret*=now;
        now*=now;
        k/=2;
    }
    return ret;
}
ll gya[200010];
ll kai[200010];
ll m;
ll beki(ll n,ll k){
  ll ret=1;
  ll now=n;
  while(k>0){
    if(k%2==1){
      ret*=now;
      ret%=m;
    }
    now*=now;
    now%=m;
    k/=2;
  }
  return ret;
}
ll gyaku(ll n){
  return beki(n,mod-2);
}
void nckinit(ll n){
  kai[0]=1;
  kai[1]=1;
  for(int i=2;i<=n;i++){
    kai[i]=kai[i-1]*i;
    kai[i]%=mod;
  }
  gya[n]=gyaku(kai[n]);
  for(int i=n-1;i>=1;i--){
    gya[i]=gya[i+1]*(i+1);
    gya[i]%=mod;
  }
  gya[0]=1;
}    
ll nck(ll n,ll k){
  if(k<0)return 0;
  if(k==0||n==k)return 1;
  ll ret=kai[n];
  ret*=gya[n-k];
  ret%=mod;
  ret*=gya[k];
  ret%=mod;
  return ret;
}
ll npk(ll n,ll k){
  if(k<0)return 0;
  if(k==0)return 1;
  ll ret=kai[n];
  ret*=gya[n-k];
  ret%=mod;
  return ret;
}
ll dp[200010];
vector<pll> v[200010];
void dfs(ll now,ll par){
  
    ll ret=0;
    rep(i,0,v[now].size()){
      ll ne=v[now][i].first;
      if(ne==par)continue;
      dfs(ne,now);
      ret+=dp[ne];
      
    }
    dp[now]=ret+1;
  
}
int main(){
  int n;cin>>n;
  rep(i,0,n-1){
    ll a,b,c;cin>>a>>b>>c;
    v[a].push_back({b,c});
    v[b].push_back({a,c});
  }
  ll d[n+1];
  fill(d,d+n+1,inf);
  d[1]=0;
  queue<ll> q;
  q.push(1);
  while(q.size()>0){
    ll now=q.front();q.pop();
    rep(i,0,v[now].size()){
      pll p=v[now][i];
      ll ne=p.first;ll co=p.second;
      if(d[ne]<inf)continue;
      d[ne]=d[now]+co;
      q.push(ne);
    }
  }
  dfs(1,-1);
  ll sum[n+1];
  fill(sum,sum+n+1,0);
  rep(i,1,n+1)sum[1]+=d[i];
  q.push(1);
  while(q.size()>0){
    ll now=q.front();q.pop();
    rep(i,0,v[now].size()){
      ll ne=v[now][i].first;
      if(sum[ne]>0)continue;
      sum[ne]=sum[now]+(n-dp[ne]*2)*v[now][i].second;
      q.push(ne);
    }
  }
  ll ans=0;
  rep(i,1,n+1)ans+=sum[i];
  cout<<ans<<endl;
  //rep(i,1,n+1)cout<<sum[i]<<" ";
}
0