結果
問題 | No.872 All Tree Path |
ユーザー |
|
提出日時 | 2019-08-30 21:52:04 |
言語 | C++11 (gcc 13.3.0) |
結果 |
AC
|
実行時間 | 428 ms / 3,000 ms |
コード長 | 1,864 bytes |
コンパイル時間 | 1,500 ms |
コンパイル使用メモリ | 171,508 KB |
実行使用メモリ | 63,256 KB |
最終ジャッジ日時 | 2024-11-21 22:43:10 |
合計ジャッジ時間 | 5,921 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
#include <bits/stdc++.h>using namespace std;#define F first#define S second#define R cin>>#define Z class#define ll long long#define ln cout<<'\n'#define in(a) insert(a)#define pb(a) push_back(a)#define pd(a) printf("%.10f\n",a)#define mem(a) memset(a,0,sizeof(a))#define all(c) (c).begin(),(c).end()#define iter(c) __typeof((c).begin())#define rrep(i,n) for(ll i=(ll)(n)-1;i>=0;i--)#define REP(i,m,n) for(ll i=(ll)(m);i<(ll)(n);i++)#define rep(i,n) REP(i,0,n)#define tr(it,c) for(iter(c) it=(c).begin();it!=(c).end();it++)template<Z A>void pr(A a){cout<<a;ln;}template<Z A,Z B>void pr(A a,B b){cout<<a<<' ';pr(b);}template<Z A,Z B,Z C>void pr(A a,B b,C c){cout<<a<<' ';pr(b,c);}template<Z A,Z B,Z C,Z D>void pr(A a,B b,C c,D d){cout<<a<<' ';pr(b,c,d);}template<Z A>void PR(A a,ll n){rep(i,n){if(i)cout<<' ';cout<<a[i];}ln;}ll check(ll n,ll m,ll x,ll y){return x>=0&&x<n&&y>=0&&y<m;}const ll MAX=1e9+7,MAXL=1LL<<61,dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};typedef pair<ll,ll> P;typedef pair<P,ll> PP;int n;vector<P> v[222222];int solve(int i,int p) {if(v[i].size()==1&&v[i][0].F==p) {v[i][0].S=n-1;return 1;}int q=0,k=-1;for(int j=0; j<v[i].size(); j++) {if(v[i][j].F!=p) {if(v[i][j].S==-1) {v[i][j].S=solve(v[i][j].F,i);q+=v[i][j].S;} else q+=v[i][j].S;} else k=j;}if(k!=-1) v[i][k].S=n-1-q;return q+1;}void Main() {cin >> n;PP p[n-1];rep(i,n-1) {ll x,y,z;cin >> x >> y >> z;x--,y--;v[x].pb(P(y,-1));v[y].pb(P(x,-1));p[i]=PP(P(x,y),z);}solve(0,-1);map<P,ll> m;rep(i,n)rep(j,v[i].size())m[P(i,v[i][j].F)]=v[i][j].S;ll ans=0;rep(i,n-1) {ll x=p[i].F.F,y=p[i].F.S,z=p[i].S;ll d=m[P(x,y)],c=m[P(y,x)];ans+=d*c*z*2;}pr(ans);}int main(){ios::sync_with_stdio(0);cin.tie(0);Main();return 0;}