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道门联通。

  每一个房间都有一个价值为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

题解：其实这个模型是一棵树， 对于树上的某一节点 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;
}