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

1 1

3 1

4 5 6

1 2 2 3

Sample Output

20 6

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<math.h>
using namespace std;
#define INF 0x3f3f3f3f
#define N 12345
int k;
int a[N],aa[N],dis[N];
struct node
{
    int e,f,v;
}q[N<<2];
void add(int e,int f)
{
  q[k].e=e;
  q[k].f=f;
  q[k].v=aa[e];
  aa[e]=k++;
}
int dfs(int s)
{
    int sum=0;
    for(int i=aa[s];i!=-1;i=q[i].v)
    {
        int f=q[i].f;
        if(!dis[f])
        {
            dis[f]=1;
            sum+=dfs(f);
        }
    }
    return max(a[s],sum);
}
int main()
{
    int T,m,n,e,f;
    scanf("%d",&T);
    while(T--)
    {
        k=0;
        memset(aa,-1,sizeof(aa));
        memset(dis,0,sizeof(dis));
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&e,&f);
            add(e,f);
            add(f,e);
        }
        dis[m]=1;
        printf("%d\n",dfs(m));
    }
    return 0;
}

这是一个比赛题一开始读懂之后整个人都感觉写不出来就果断没有写（并且自己精力不足）就写了一个字符串题最后字符串的写出来了自己也写得烦烦的后来看了题解才这道也是水题

zzuli 1907: 小火山的宝藏收益

标签：

原文地址：http://www.cnblogs.com/a719525932/p/5763959.html