結果

問題 No.872 All Tree Path
ユーザー kenta255kenta255
提出日時 2019-08-30 23:09:28
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 280 ms / 3,000 ms
コード長 1,852 bytes
コンパイル時間 2,192 ms
コンパイル使用メモリ 206,412 KB
実行使用メモリ 28,872 KB
最終ジャッジ日時 2023-08-14 07:50:18
合計ジャッジ時間 5,918 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 280 ms
24,588 KB
testcase_01 AC 279 ms
24,684 KB
testcase_02 AC 270 ms
24,684 KB
testcase_03 AC 187 ms
28,872 KB
testcase_04 AC 4 ms
9,264 KB
testcase_05 AC 276 ms
24,552 KB
testcase_06 AC 257 ms
24,512 KB
testcase_07 AC 273 ms
24,516 KB
testcase_08 AC 23 ms
10,880 KB
testcase_09 AC 23 ms
10,832 KB
testcase_10 AC 23 ms
10,848 KB
testcase_11 AC 23 ms
10,852 KB
testcase_12 AC 24 ms
10,852 KB
testcase_13 AC 4 ms
9,260 KB
testcase_14 AC 4 ms
9,316 KB
testcase_15 AC 4 ms
9,316 KB
testcase_16 AC 4 ms
9,604 KB
testcase_17 AC 4 ms
9,308 KB
testcase_18 AC 4 ms
9,320 KB
testcase_19 AC 4 ms
9,344 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