Description
在X森林里，上帝创建了生命之树。
他给每棵树的每个节点（叶子也称为一个节点）上，都标了一个整数，代表这个点的和谐值。
上帝要在这棵树内选出一个非空节点集S，使得对于S中的任意两个点a,b，都存在一个点列 {a, v1, v2, …, vk, b} 使得这个点列中的每个点都是S里面的元素，且序列中相邻两个点间有一条边相连。
在这个前提下，上帝要使得S中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过atm的努力，他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算，他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。
「输入格式」
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数，依次表示每个节点的评分。
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。由于这是一棵树，所以是不存在环的。
「输出格式」
输出一行一个数，表示上帝给这棵树的分数。
「样例输入」
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
「样例输出」
8
Input
第一行一个整数 n 表示这棵树有 n 个节点。
第二行 n 个整数，依次表示每个节点的评分。
接下来 n-1 行，每行 2 个整数 u, v，表示存在一条 u 到 v 的边。由于这是一棵树，所以是不存在环的。
Output
输出一行一个数，表示上帝给这棵树的分数。
Sample Input
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
Sample Output
8
这题瞎搜，居然被我水过去了，严重怀疑学校的测试数据：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
| package 省赛2015;
import java.util.Scanner;
public class T9 {
static boolean[][] data;
static boolean[][] data1;
static int[] fenzhi;
static int max;
static int n;
static boolean[] vis;
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
n=sc.nextInt();
fenzhi=new int[n+1];
for (int i = 1; i <= n; i++) {
fenzhi[i]=sc.nextInt();
}
data=new boolean[n+1][n+1];
data1=new boolean[n+1][n+1];
vis=new boolean[n+1];
for (int i = 1; i < n; i++) {
int a=sc.nextInt();
int b=sc.nextInt();
data1[a][b]=data1[b][a]=data[a][b]=data[b][a]=true;
}
//
for (int i = 1; i < n; i++) {
for (int j = i+1; j <= n; j++) {
vis=new boolean[n+1];
vis[i]=true;
dfs(i,j,fenzhi[i]);
for (int a = 0; a <= n; a++) {
for (int b = 0; b <= n; b++) {
data[a][b]=data1[a][b];
}
}
}
}
System.out.println(max);
}
private static void dfs(int begin,int end,int fen) {
if(data[begin][end]){
for (int i = 1; i <= n; i++) {
if(vis[i]||i==end)continue;
if(data[begin][i]){
if(fenzhi[i]>0)
fen+=fenzhi[i];
}
}
for (int i = 1; i <= n; i++) {
if(vis[i]||i==begin)continue;
if(data[end][i]){
if(fenzhi[i]>0)
fen+=fenzhi[i];
}
}
fen+=fenzhi[end];
if(fen>max){
max=fen;
}
return;
}
for (int i = 1; i <= n; i++) {
if(vis[i])continue;
if(data[end][i]){
data[end][i]=false;
vis[i]=true;
fen+=fenzhi[i];
dfs(end,i,fen);
data[end][i]=true;
vis[i]=false;
fen-=fenzhi[i];
}
}
}
}
|