题目描述
在 X 森林里,上帝创建了生命之树。
他给每棵树的每个节点(叶子也称为一个节点)上,都标了一个整数,代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集 S,使得对于 S 中的任意两个点 a,b,都存在一个点列 a,v1,v2,⋯,vk,b 使得这个点列中的每个点都是 S 里面的元素,且序列中相邻两个点间有一条边相连。
在这个前提下,上帝要使得 S 中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过 atm 的努力,他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算,他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。
输入描述
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数,依次表示每个节点的评分。
接下来 n−1 行,每行 2 个整数 u,v,表示存在一条 u 到 v 的边。由于这是一棵树,所以是不存在环的。
其中,0<n≤10^5, 每个节点的评分的绝对值不超过 10^6。
输出描述
输出一行一个数,表示上帝给这棵树的分数。
输入输出样例
示例
输入
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
输出
8
这题利用了动态规划和dfs,题解如下:
import java.math.*;
import java.util.*;
public class Main{
static int n;
static int [][]road;//存储两节点之间是否有通路
static boolean []vis;//是否已被访问
static int []value;//每个节点的评分
static int max=0;
public static void dfs(int m) {
vis[m]=true;
for(int i=1;i<=n;i++) {
if(vis[i]==false&&road[m][i]!=0) {//当节点之间有通路并且还没有被访问
dfs(i);
value[m]=Math.max(value[m], value[m]+value[i]);
max=Math.max(max, value[m]);
}
}
}
public static void main(String[] args) {
Scanner in=new Scanner(System.in);
n=in.nextInt();
road=new int[n+1][n+1];
value=new int[n+1];
vis=new boolean[n+1];
for(int i=1;i<=n;i++)
value[i]=in.nextInt();
for(int i=1;i<=n-1;i++) {
int u=in.nextInt();
int v=in.nextInt();
road[u][v]=1;
road[v][u]=1;
}
dfs(1);
System.out.println(max);
}
}