题目描述
在 X 森林里,上帝创建了生命之树。
他给每棵树的每个节点(叶子也称为一个节点)上,都标了一个整数,代表这个点的和谐值。
上帝要在这棵树内选出一个节点集合 �(允许为空集),使得对于 � 中的任意两个点 �,�,都存在一个点列 �,�1,�2,⋯,��,� 使得这个点列中的每个点都是 � 里面的元素,且序列中相邻两个点间有一条边相连。
在这个前提下,上帝要使得 � 中的点所对应的整数的和尽量大。
这个最大的和就是上帝给生命之树的评分。
经过 atm 的努力,他已经知道了上帝给每棵树上每个节点上的整数。但是由于 atm 不擅长计算,他不知道怎样有效的求评分。他需要你为他写一个程序来计算一棵树的分数。
输入格式
第一行一个整数 � 表示这棵树有 � 个节点。
第二行 � 个整数,依次表示每个节点的评分。
接下来 �−1 行,每行 2 个整数 �,�,表示存在一条 � 到 � 的边。由于这是一棵树,所以是不存在环的。
输出格式
输出一行一个数,表示上帝给这棵树的分数。
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<string>
#include<vector>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define N 100005
int val[N],vis[N],dp[N][2];
int n,a,b,ans=-0x3f3f3f3f;
vector<int> node[N];
void dfs(int u)
{
dp[u][1]=val[u];
dp[u][0]=0;
vis[u]=1;
for(int i=0;i<node[u].size();i++)
{
if(!vis[node[u][i]])
{
dfs(node[u][i]);
dp[u][1]+=max(dp[node[u][i]][1],dp[node[u][i]][0]);
}
else
{
dp[u][1]=max(dp[u][1],val[u]);
dp[u][0]=max(dp[u][0],0);
}
}
}
int main()
{
cin>>n;
int i;
for(i=1;i<=n;i++)
cin>>val[i];
for(i=1;i<n;i++)
{
cin>>a>>b;
node[a].push_back(b);
node[b].push_back(a);
}
memset(vis,0,sizeof(vis));
memset(dp,0,sizeof(dp));
dfs(1);
for(i=1;i<=n;i++)
ans=max(ans,dp[i][1]);
cout<<ans<<endl;
return 0;
}