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
21 1203 14 5 61 22 3
Sample Output
206
和1900题 类似。 用邻接表。
然后遍历一遍,找出最优解。
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define LL long long
#define N 10010
#define mod 1000000007
int head[N+N],v[N],b[N],t;
struct node
{
int to;
int w;
int next;
}a[N+N];
void add(int to,int w)
{
a[t].to=to;
a[t].w=w;
a[t].next=head[to];
head[to]=t++;
}
int dfs(int x)
{
v[x]=1;
int sum=0,maxx=b[x];
for(int i=head[x];i!=-1;i=a[i].next)
{
int wi=a[i].w;
if(!v[wi])
sum+=dfs(wi);
}
return max(sum,maxx);
}
int main()
{
int T,n,k,x,y;
cin>>T;
while(T--)
{
int ans=0,s;
t=0;
memset(head,-1,sizeof(head));
memset(v,0,sizeof(v));
cin>>n>>s;
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
for(int i=0;i<n-1;i++){
scanf("%d%d",&x,&y);
add(x,y);
add(y,x);
}
printf("%d\n",dfs(s));
}
return 0;
}