Submission #1864837

---

Source Code Expand

#include<cstdio>
#include<algorithm>
#include<vector>
#include<deque>
#include<stack>
#include<queue>
#include<string>
#include<iostream>
#include<tuple>
#include<utility>
#include<set>
#include<queue>
#include<iomanip>
#include<iterator>
#include<map>
#include<bitset>
#include<array>
#include<cmath>
#include<functional>
#include<list>
using namespace std;
typedef long long int llint;
typedef long double lldo;
const llint big=2e18+100000;
const int mod=1e9+7;
const lldo eps=1e-7;
const long double pai=3.141592653589793238462643383279502884197;
#define mt make_tuple
#define mp make_pair
#define fir first
#define sec second
#define pub push_back
#define puf push_front
#define pob pop_back
#define pof pop_front
#define res resize
#define ins insert
#define era erase
#define dme(in) cout<<in<<endl;return 0
template <class T,class U>void mineq(T& a,U b){if(a>b){a=b;}}
template <class T,class U>void maxeq(T& a,U b){if(a<b){a=b;}}
int main(void){
	int i,n;cin>>n;
	vector<tuple<llint,int,int>>hen(n-1);
	vector<int>oya(n);
	vector<int>siz(n,1);
	vector<llint>imo(n);//下方向
	for(i=0;i<n;i++){oya[i]=i;}
	for(i=0;i<n-1;i++){
		int a,b,c;cin>>a>>b>>c;a--;b--;
		hen[i]=mt(c,a,b);
	}
	sort(hen.rbegin(),hen.rend());
	//すぬけ 木である必要はない
	for(i=0;i<n-1;i++){
		int a,b;llint c;
		tie(c,a,b)=hen[i];
		while(a!=oya[a]){a=oya[a];}
		while(b!=oya[b]){b=oya[b];}
		if(a==b){continue;}
		if(siz[a]<siz[b]){swap(a,b);}
		imo[a]+=c*siz[b];
		imo[b]+=c*siz[a]-imo[a];
		siz[a]+=siz[b];
		oya[b]=a;
	}
	vector<pair<int,int>>miru(n);
	for(i=0;i<n;i++){miru[i]=mp(siz[i],i);}
	//子のほうがサイズが小さいので親から見る=サイズの大きい順
	sort(miru.rbegin(),miru.rend());
	for(i=1;i<n;i++){//最大は根なので
		imo[miru[i].sec]+=imo[oya[miru[i].sec]];
	}
	for(i=0;i<n;i++){cout<<imo[i]<<endl;}
	return 0;
}

Submission Info

Judge Result