2018四川省程序设计竞赛 I. Island(Java版)

题目链接：https://www.oj.swust.edu.cn/problem/show/2812
题目大意：给你一个树，每个点有一定的权值，删除某个节点极其与其他节点相连的边会形成几棵新的树，这些树中权值和最大的是你会收到的伤害，你希望知道收到的最小伤害是多少，也就是遍历所有节点求最大权值的最小值。 具体看代码注释
AC代码：

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.StreamTokenizer;
//画个图比较容易理解，我们按照题意，站在一个点上之后，所有和该点相连的边断掉，求新的几个树的最大权值，相当于求
//原先树的权值和减去该点极其子树的权值和，以及该点的所有子树中的最大值，存储这个最大值，然后遍历一遍所有的点，
//找出最大值的最小值

//把1当做根节点，两遍dfs，第一遍dfs求该节点和其子树权值总和，存入val[]，将该节点所有子树中最大权值存入M[].
//第二遍dfs，遍历每个点，比较M[i]和总权值与该点总权值的差（val[1]-val[i]）,将较大的值存入ans[]
public class Main {
	static int maxn=(int) (2e5+10);
	static class edge{
		int v,next;

		public edge(int v, int next) {
			super();
			this.v = v;
			this.next = next;
		}
		
	}
	static edge e[]=new edge[maxn*2];
	static int p[]=new int[maxn*2];
	static long cost[]=new long[maxn];//存放节点的初值
	static long ans[]=new long[maxn];//存放M[i]与val[1]-val[i]的最大值
	static boolean vis[]=new boolean[maxn];//标记是否访问该节点
	static long M[]=new long[maxn];//存储该节点最大的子树的权值
	static long val[]=new long[maxn];//存储该节点极其子树的权值和
	static int n,T,u,v,eid;
	static void init(){
		for(int i=0;i<=2*n+10;i++)
			p[i]=-1;
		for(int i=1;i<=n;i++){
			vis[i]=false;
			cost[i]=0;
			ans[i]=0;
			M[i]=0;
			val[i]=0;
		}
		eid=0;
	}
	static void insert(int u,int v){
		e[eid]=new edge(v,p[u]);
		p[u]=eid++;
	}
	static void dfs1(int u){
		M[u]=(long) -(1e18+10);
		for(int i=p[u];i!=-1;i=e[i].next){
			int v=e[i].v;
			if(!vis[v]){
				vis[v]=true;
				dfs1(v);
				val[u]+=val[v];
				M[u]=Math.max(M[u], val[v]);
			}
		}
		val[u]+=cost[u];
	}
	static void dfs2(int u){
		for(int i=p[u];i!=-1;i=e[i].next){
			int v=e[i].v;
			if(!vis[v]){
				vis[v]=true;
				dfs2(v);
			}
		}
		ans[u]=Math.max(M[u], val[1]-val[u]);
	}
	static StreamTokenizer st=new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
	public static void main(String[] args) {
		T=nextInt();
		while(T-->0){
			n=nextInt();
			long Min=(long) (1e18+10);
			init();
			for(int i=1;i<=n;i++){
				cost[i]=nextInt();
			}
			for(int i=0;i<n-1;i++){
				u=nextInt();
				v=nextInt();
				insert(u,v);
				insert(v,u);
			}
			vis[1]=true;
			dfs1(1);
			for(int i=2;i<=n;i++){
				vis[i]=false;
			}
			dfs2(1);
			for(int i=1;i<=n;i++){
				Min=Math.min(Min, ans[i]);
			}
			System.out.println(Min);
		}
	}
	static int nextInt(){
		try {
			st.nextToken();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return (int)st.nval;
	}
}