結果

問題 No.872 All Tree Path
ユーザー kenta255kenta255
提出日時 2019-08-30 23:09:28
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 248 ms / 3,000 ms
コード長 1,852 bytes
コンパイル時間 2,423 ms
コンパイル使用メモリ 209,224 KB
実行使用メモリ 29,676 KB
最終ジャッジ日時 2024-11-22 02:47:26
合計ジャッジ時間 5,467 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 244 ms
25,156 KB
testcase_01 AC 245 ms
25,516 KB
testcase_02 AC 248 ms
25,816 KB
testcase_03 AC 203 ms
29,676 KB
testcase_04 AC 4 ms
10,468 KB
testcase_05 AC 245 ms
25,360 KB
testcase_06 AC 243 ms
24,920 KB
testcase_07 AC 243 ms
25,256 KB
testcase_08 AC 24 ms
12,244 KB
testcase_09 AC 24 ms
11,696 KB
testcase_10 AC 24 ms
12,772 KB
testcase_11 AC 24 ms
11,332 KB
testcase_12 AC 24 ms
11,856 KB
testcase_13 AC 4 ms
10,368 KB
testcase_14 AC 4 ms
9,664 KB
testcase_15 AC 4 ms
9,796 KB
testcase_16 AC 4 ms
9,796 KB
testcase_17 AC 3 ms
9,924 KB
testcase_18 AC 4 ms
9,408 KB
testcase_19 AC 4 ms
9,924 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define P pair<ll,ll>
#define FOR(I,A,B) for(ll I = ll(A); I < ll(B); ++I)
#define FORR(I,A,B) for(ll I = ll((B)-1); I >= ll(A); --I)
#define TO(x,t,f) ((x)?(t):(f))
#define SORT(x) (sort(x.begin(),x.end())) // 0 2 2 3 4 5 8 9
#define POSL(x,v) (lower_bound(x.begin(),x.end(),v)-x.begin()) //xi>=v  x is sorted
#define POSU(x,v) (upper_bound(x.begin(),x.end(),v)-x.begin()) //xi>v  x is sorted
#define NUM(x,v) (POSU(x,v)-POSL(x,v))  //x is sorted
#define REV(x) (reverse(x.begin(),x.end())) //reverse
ll gcd(ll a,ll b){if(a%b==0)return b;return gcd(b,a%b);}
ll lcm(ll a,ll b){ll c=gcd(a,b);return ((a/c)*(b/c)*c);}
#define NEXTP(x) next_permutation(x.begin(),x.end())
const ll INF=ll(1e18)+ll(7);
const ll MOD=1000000007LL;
#define out(a) cout<<fixed<<setprecision((a))

vector<ll> tree[200003];
ll ans = 0;
ll N;
ll root = 1;
ll pa[200009]={};
ll numch[200009]={};
bool did[200009];
vector<ll> U,V,W;

void dfs(int i = 1){
	did[i] = true;
	for(auto ch:tree[i]){
		if(did[ch])continue;
		pa[ch] = i;
		dfs(ch);
	}
}

int main(){
	cin >> N;
	FOR(i,0,N-1){
		ll u,v,w;
		cin >> u >> v >> w;
		tree[u].push_back(v);
		tree[v].push_back(u);
		U.push_back(u);
		V.push_back(v);
		W.push_back(w);
	}
	dfs();
	FOR(i,1,N+1) did[i] = false;
	vector<ll> dodo;
	queue<ll> q;
	q.push(1);
	did[1] = true;
	while(q.size()){
		ll ind = q.front();
		q.pop();
		dodo.push_back(ind);
		for(auto nex:tree[ind]){
			if(not did[nex]){
				did[nex] = true;
				q.push(nex);
			}
		}
	}
	reverse(dodo.begin(), dodo.end());
	//cout << dodo.size() << endl;
	ll cnt[N+1] = {};
	FOR(i,0,N+1)cnt[i]=1;
	for(auto x:dodo){
		cnt[pa[x]]+=cnt[x];
	}
	FOR(i,0,N-1){
		ll chai;
		if(pa[U[i]]==V[i]){
			chai = U[i];
		}else{
			chai = V[i];
		}
		ans += W[i] * cnt[chai] * (N-cnt[chai]);
	}
	cout << ans*2 << endl;
}
0