zzulioj-1907-小火山的宝藏收益【DFS】

题目链接：​​点击打开链接​​

1907: 小火山的宝藏收益

Time Limit: 1 Sec   Memory Limit: 128 MB

Submit: 179  

Solved: 53

​​Submit​​

​​Status​​

​​Web Board​​

Description

  

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

也就是说后面的宝藏小火山是得不到了（进入这个房间的门是不会关闭的，小火山还是可以回去的）；如果小火山不取这个宝藏，而是去打开通往另一房间的门，那么这个房间的宝藏就会消失， 小火山就得不到这个房间的宝藏。

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

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

题解：其实这个模型是一棵树， 对于树上的某一节点 i 点，那么对于 i 点有两种选择：选择该点价值或者进入该点的子节点。注意题中标注，如上图：拿点 4 来说，在小火山到达点 4 时，存在两种情况：1、小火山可能取点4的宝藏，那么点 8、9、10 的宝藏就不能得到了；2、不取点4的宝藏去取点8、9、10加在一起的宝藏，就是说在点 4 能取得的宝藏为 max（val [ 4 ]，val [ 8 ] + val [ 9 ] + val [ 10 ] ）；

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAX=1e4+10;
int N,S;
struct node
{
  int from,to,next;
}edge[2*MAX]; // 刚开始开的是 MAX 一连 WA 了好几次，结果oj上就说 WA ，没给数组越界提示，害得浪费了一个小时，才看出来

int num;
int val[MAX];
int head[MAX];
bool vis[MAX];
void init()
{
  memset(head,-1,sizeof(head));
  memset(val,0,sizeof(val));
  memset(vis,0,sizeof(vis));
  num=0;
}

void add(int a,int b)
{
  edge[num].from=a;
  edge[num].to=b;
  edge[num].next=head[a];
  head[a]=num++;
}

int DFS(int x)
{
  vis[x]=1;
  int sum=0;
  int now=val[x];
  for(int i=head[x];i!=-1;i=edge[i].next)
  {
    int id=edge[i].to;
    if(!vis[id])
    {
      sum+=DFS(id);
      vis[id]=1;
    }
  }
  return now>sum?now:sum;
}

int main()
{
  int t;
  scanf("%d",&t);
  while(t--)
  {
    scanf("%d %d",&N,&S);
    init();
    for(int i=1;i<=N;i++)
      scanf("%d",&val[i]);
    int a,b;
    for(int i=1;i<=N-1;i++)
    {
      scanf("%d %d",&a,&b);
      add(a,b);
      add(b,a);
    }
    printf("%d\n",DFS(S));
  }
  return 0;
}