zzuli 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

21 1203 14 5 61 22 3

Sample Output

206

和1900题 类似。 用邻接表。

然后遍历一遍，找出最优解。

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;

#define LL long long
#define N 10010
#define mod 1000000007
int head[N+N],v[N],b[N],t;
struct node
{
    int to;
    int w;
    int next;
}a[N+N];
void add(int to,int w)
{
    a[t].to=to;
    a[t].w=w;
    a[t].next=head[to];
    head[to]=t++;
}
int dfs(int x)
{
    v[x]=1;
    int sum=0,maxx=b[x];
    for(int i=head[x];i!=-1;i=a[i].next)
    {
        int wi=a[i].w;
        if(!v[wi])
            sum+=dfs(wi);
    }
    return max(sum,maxx);
}
int main()
{
    int T,n,k,x,y;
    cin>>T;
    while(T--)
    {
        int ans=0,s;
        t=0;
        memset(head,-1,sizeof(head));
        memset(v,0,sizeof(v));
        cin>>n>>s;
        for(int i=1;i<=n;i++)
            scanf("%d",&b[i]);
        for(int i=0;i<n-1;i++){
            scanf("%d%d",&x,&y);
            add(x,y);
            add(y,x);
        }
        printf("%d\n",dfs(s));
    }
    return 0;
}