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
#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 ",dfs(m)); } return 0; }