Submission #3321176

---

Source Code Expand

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef long long ll;
int N,A[100010],B[100010],E[100010] = {0},visited[200010] = {0};
ll C[100010],ans[200010] = {0};

vector<pair<ll,int>> e;
vector<vector<pair<ll,int>>> v(200010);

int p[100010] = {0},r[100010] = {0};
ll s[100010] = {0};

void init(int N){
	for(int i=0;i<=N;++i){
		p[i] = i;
		s[i] = 1;
		E[i] = i;
	}
}

int root(int a){
	if(p[a] == a) return a;
	return (p[a] = root(p[a]));
}

bool is_same_set(int a,int b){
	return root(a) == root(b);
}

void unite(int a, int b){
	a = root(a);
	b = root(b);
	if(a==b) return;
	if(r[a]<r[b]){
		p[a] = b;
		s[b] += s[a];
		s[a] = s[b];
	}else{
		p[b] = a;
		s[a] += s[b];
		s[b] = s[a];
		if(r[a] == r[b]) r[a]++;
	}
}

void dfs(int n){
	visited[n] = 1;
	for(int i=0;i<v[n].size();i++){
		int t = v[n][i].second;
		ll cost = v[n][i].first;
		if(visited[t]==0){
			ans[t] = ans[n] + cost;
			dfs(t);
		}
	}
}

int main(){
	cin >> N;
	init(N);
	if(N==1){
		cout << 0 << endl;
		return 0;
	}
	for(int i=1;i<=N-1;i++){
		cin >> A[i] >> B[i] >> C[i];
		e.push_back({C[i],i});
	}
	sort(e.begin(),e.end());
	for(int i=N-2;i>=0;i--){
		int t = e[i].second;
		ll cost = e[i].first;
		v[t+N].push_back({cost*s[root(B[t])],E[root(A[t])]});
		v[t+N].push_back({cost*s[root(A[t])],E[root(B[t])]});
		unite(A[t],B[t]);
		E[root(A[t])] = t+N;
	}
	dfs(N+e[0].second);
	for(int i=1;i<=N;i++){
		cout << ans[i] << endl;
	}
}

Submission Info

Judge Result