ZZULIoj 1907 小火山的宝藏收益

Description

进去宝藏后， 小火山发现宝藏有N个房间，且这n个房间通过N-1道门联通。

每一个房间都有一个价值为Ai的宝藏， 但是每一个房间也都存在一个机关。如果小火山取走了这个房间的宝藏，那么这个房间通往其他房间的门就永远打不开了，也就是说后面的宝藏小火山是得不到了（进入这个房间的门是不会关闭的，小火山还是可以回去的）；如果小火山不取这个宝藏，而是去打开通往另一房间的门，那么这个房间的宝藏就会消失， 小火山就得不到这个房间的宝藏。

不过，小火山已经有了藏宝图，知道每一个房间的宝藏的价值，现在想请你帮小火山算一下，他最多能获得多少钱去买股票？

Input

输入第一行是一个整数T（T <= 50), 表示一共有T组数据。

对于每一组数据，第一行是两个数N, S（1 <= N <= 10000, 1 <= S <= N), N代表有N个房间， S代表小火山进去宝藏后的

起始房间（小火山怎么进入起始房间不重要），第二行是N个数，代表每个房间宝藏的价值， 随后N-1行， 每行两个数A, B， 代表

A, B这两个房间联通。

Output

对于每一组数据输出一个整数， 代表小火山能获得的最大钱数。

Sample Input

2
1 1
20
3 1
4 5 6
1 2
2 3

Sample Output

20
6

树的模型，从树根开始出发，比较树根的值和树根所有直接相连的子节点的最大收益的和，输出大的值，求树根所有直接相连的子节点的最大收益和，要先求出每个子节点的最大收益，把每个子节点看成树根的问题，求法一样，用函数递归。例如根节点为1，先比较1的值个2，5最大收益和，求2的最大收益要比较2的值和3,4的最大收益和，因为3,4没有子节点，所以最大收益为本身，所以2的子节点的最大收益和为5比2本身的值4大，所以2节点的最大收益为5，同样比较5节点最大收益为3，所以2,3的最大收益和为8小于1节点的值，1节点的值就是最大收益。

 1 #include<cstdio>
 2 #include<string.h>
 3 int n,m,head[20010],flag[20010];
 4 int num,mon[20010];
 5 struct stu
 6 {
 7     int from,to,next;
 8 }edge[20010];
 9 void init()
10 {
11     memset(head,-1,sizeof(head));
12     memset(flag,0,sizeof(flag));
13     num=0;
14 }
15 void add_edge(int u,int v)
16 {
17     edge[num].from=u;
18     edge[num].to=v;
19     edge[num].next=head[u];
20     head[u]=num++;
21 }
22 int dfs(int u)                        //函数作用为返回 max（当前节点的值，当前节点的子节点值的和）
23 {
24     flag[u]=1;
25     int money=mon[u];                //记录当前节点的值
26     int ans=0,i,v;                    //ans表示子节点的和
27     for(i = head[u] ; i != -1 ; i=edge[i].next)
28     {
29         v=edge[i].to;
30         if(!flag[v])
31         {
32             ans+=dfs(v);            //求子节点值的和，dfs返回 max（子节点值，子节点的字节的值的和）
33             flag[v]=1;
34         }
35     }
36     return ans=ans>money?ans:money; //返回当前节点和子节点值较大的
37 }
38 int main()
39 {
40     int t;
41     scanf("%d",&t);
42     while(t--)
43     {
44         int a,b,i;
45         int n,s,ans;
46         scanf("%d %d",&n,&s);
47         init();
48         for(i = 1 ; i <= n ; i++)
49         {
50             scanf("%d",&mon[i]);
51         }
52         for(i = 0 ; i < n-1 ; i++)
53         {
54             scanf("%d %d",&a,&b);
55             add_edge(a,b);
56             add_edge(b,a);
57         }
58         ans=dfs(s);
59         printf("%d\n",ans);
60     }
61  }