zzuli OJ-1107 小火山的宝藏收藏（dfs+邻接表）



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

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<limits.h>
#include<stdlib.h>
#include<math.h>
#include<queue>
#include<map>
#define maxn 100005
using namespace std;
int n,m,cnt,head[maxn];
long long ans,a[maxn],vis[maxn];
struct  node
{
	int u;
	int v;
	int w;
	int next;
}edge[maxn];
void build(int u,int v)
{
	edge[cnt].u=u;
	edge[cnt].v=v;
	edge[cnt].w=1;
	edge[cnt].next=head[u];
	head[u]=cnt++;
}
int dfs(int u)
{
	int i,k=0;
	long long sum=0;
	for(i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;
		if(!vis[v])
		{
			vis[v]=1;
			k=1;
			sum+=dfs(v);
		}
	}
	if(!k)
	  return a[u];
	else
	   return max(a[u],sum);
}
int main()
{
	int T,i,j,x,y;
	scanf("%d",&T);
	while(T--)
	{
		memset(head,-1,sizeof(head));
		memset(vis,0,sizeof(vis));
		cnt=0;
		scanf("%d%d",&n,&m);
		for(i=1;i<=n;i++)
		   scanf("%d",&a[i]);
		for(i=1;i<=n-1;i++)
		{
			scanf("%d%d",&x,&y);
			build(x,y);
			build(y,x);
		}
		vis[m]=1;
		ans=dfs(m);
		printf("%lld\n",ans);
	}
}

 2 3

Sample Output

20 6