Submission #3218558

---

Source Code Expand

import java.io.IOException; 
import java.io.InputStream; 
import java.io.PrintWriter; 
import java.util.*; 
 

class Main{

	static class Edge{
		int from, to;
		long cost;
		Edge(int f, int t, long c){from=f;to=t;cost=c;}
	}

	static class UF{
		int[] pare;
		Node[] pareNode;
		UF(int n){
			pare =new int[n];
			pareNode=new Node[n];
			Arrays.fill(pare, -1);
			for(int i=0;i<n;++i){
				pareNode[i]=new Node();
				pareNode[i].v=i;
			}
		}
		int root(int v){
			if(pare[v]<0)return v;
			return pare[v]=root(pare[v]);
		}
		boolean same(int u, int v){
			return root(u)==root(v);
		}
		int size(int v){
			return -pare[root(v)];
		}
		void unite(int u, int v, long c){
			u=root(u);v=root(v);
			if(u==v)return;
			if(pare[u]<pare[v]){
				int tmp=u;u=v;v=tmp;
			}
			Node node = new Node();
			node.n1 = pareNode[v];
			node.c1 = c*size(u);
			node.n2 = pareNode[u];
			node.c2 = c*size(v);
			pare[v]+=pare[u];
			pare[u]=v;
			pareNode[v]=node;
		}
	}
	static class Node{
		int v=-1;
		Node n1, n2;
		long c1, c2;
	}
	static long dis[];
	static void dfs(Node node, long d){
		if(node.v>=0)dis[node.v]=d;
		else{
			dfs(node.n1, d+node.c1);
			dfs(node.n2, d+node.c2);
		}
	}

	static void solve(){ 
		int n = ni();
		PriorityQueue<Edge> que = new PriorityQueue<>((a,b)->b.cost-a.cost<0 ? -1:1);
		for(int i=0;i<n-1;++i){
			int a=ni()-1, b=ni()-1;
			long c = nl();
			que.add(new Edge(a, b, c));
		}
		UF uf= new UF(n);
		while(!que.isEmpty()){
			Edge e = que.poll();
			uf.unite(e.from, e.to, e.cost);
		}
		dis = new long[n];
		Node root = uf.pareNode[uf.root(0)];
		dfs(root, 0);
		for(int i=0;i<n;++i)out.println(dis[i]);

	} 
 
 
 
 
	 public static void main(String[] args){ 
		 solve(); 
		 out.flush(); 
	 } 
	 private static InputStream in = System.in; 
	 private static PrintWriter out = new PrintWriter(System.out); 
 
	 static boolean inrange(int y, int x, int h, int w){ 
		 return y>=0 && y<h && x>=0 && x<w; 
	 } 
	 @SuppressWarnings("unchecked") 
	 static<T extends Comparable> int lower_bound(List<T> list, T key){ 
		 int lower=-1;int upper=list.size(); 
		 while(upper - lower>1){ 
		 int center =(upper+lower)/2; 
		 if(list.get(center).compareTo(key)>=0)upper=center; 
		 else lower=center; 
		 } 
		 return upper; 
	 } 
	 @SuppressWarnings("unchecked") 
	 static <T extends Comparable> int upper_bound(List<T> list, T key){ 
		 int lower=-1;int upper=list.size(); 
		 while(upper-lower >1){ 
		 int center=(upper+lower)/2; 
		 if(list.get(center).compareTo(key)>0)upper=center; 
		 else lower=center; 
		 } 
		 return upper; 
	 } 
	 @SuppressWarnings("unchecked") 
	 static <T extends Comparable> boolean next_permutation(List<T> list){ 
		 int lastIndex = list.size()-2; 
		 while(lastIndex>=0 && list.get(lastIndex).compareTo(list.get(lastIndex+1))>=0)--lastIndex; 
		 if(lastIndex<0)return false; 
		 int swapIndex = list.size()-1; 
		 while(list.get(lastIndex).compareTo(list.get(swapIndex))>=0)swapIndex--; 
		 T tmp = list.get(lastIndex); 
		 list.set(lastIndex++, list.get(swapIndex)); 
		 list.set(swapIndex, tmp); 
		 swapIndex = list.size()-1; 
		 while(lastIndex<swapIndex){ 
		 tmp = list.get(lastIndex); 
		 list.set(lastIndex, list.get(swapIndex)); 
		 list.set(swapIndex, tmp); 
		 ++lastIndex;--swapIndex; 
		 } 
		 return true; 
	 } 
	 private static final byte[] buffer = new byte[1<<15]; 
	 private static int ptr = 0; 
	 private static int buflen = 0; 
	 private static boolean hasNextByte(){ 
		 if(ptr<buflen)return true; 
		 ptr = 0; 
		 try{ 
			 buflen = in.read(buffer); 
		 } catch (IOException e){ 
			 e.printStackTrace(); 
		 } 
		 return buflen>0; 
	 } 
	 private static int readByte(){ if(hasNextByte()) return buffer[ptr++]; else return -1;} 
	 private static boolean isSpaceChar(int c){ return !(33<=c && c<=126);} 
	 private static int skip(){int res; while((res=readByte())!=-1 && isSpaceChar(res)); return res;} 
 
	 private static double nd(){ return Double.parseDouble(ns()); } 
	 private static char nc(){ return (char)skip(); } 
	 private static String ns(){ 
		 StringBuilder sb = new StringBuilder(); 
		 for(int b=skip();!isSpaceChar(b);b=readByte())sb.append((char)b); 
		 return sb.toString(); 
	 } 
	 private static int[] nia(int n){ 
		 int[] res = new int[n]; 
		 for(int i=0;i<n;++i)res[i]=ni(); 
		 return res; 
	 } 
	 private static long[] nla(int n){ 
		 long[] res = new long[n]; 
		 for(int i=0;i<n;++i)res[i]=nl(); 
		 return res; 
	 } 
	 private static int ni(){ 
		 int res=0,b; 
		 boolean minus=false; 
		 while((b=readByte())!=-1 && !((b>='0'&&b<='9') || b=='-')); 
		 if(b=='-'){ 
			 minus=true; 
			 b=readByte(); 
		 } 
		 for(;'0'<=b&&b<='9';b=readByte())res=res*10+(b-'0'); 
		 return minus ? -res:res; 
	 } 
	 private static long nl(){ 
		 long res=0,b; 
		 boolean minus=false; 
		 while((b=readByte())!=-1 && !((b>='0'&&b<='9') || b=='-')); 
		 if(b=='-'){ 
			 minus=true; 
			 b=readByte(); 
		 } 
		 for(;'0'<=b&&b<='9';b=readByte())res=res*10+(b-'0'); 
		 return minus ? -res:res; 
	} 
} 

Submission Info

Judge Result