【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

#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
struct node
{
    int from,to,next;
}st[22000];//储存临界表
int pa[11000];//数组开小了会在zzulioj上越界WA了两次
int head[11000];
int N,S;
int num;
void vis(int x,int y)
{
    node E={x,y,head[x]};//建立临界表
    st[num]=E;
    head[x]=num++;
}
int BFS(int fa ,int fb)
{
    int sum=0;
    int i;
    for(i=head[fa];i!=-1;i=st[i].next)
    {
        if(st[i].to!=fb)//防止回溯，只能向下搜索字节点，不能向上搜索父节点
            sum+=BFS(st[i].to,fa);
    }
    sum=max(sum,pa[fa]);//判断子节点值是否大于父节点
    return sum;
}
int main()
{
    int T;
    int i;
    int a,b;
    int ans;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&N,&S);
        for(i=1;i<=N;i++)
        {
            scanf("%d",&pa[i]);
            head[i]=-1;
        }
            num=0;
            for(i=1;i<N;i++)
            {
                scanf("%d%d",&a,&b);
                vis(a,b);//双向建立临界表
                vis(b,a);
            }
            ans=BFS(S,-1);
            printf("%d\n",ans);
    }
    return 0;
}