树形DP1:285
作者:
总打瞌睡的天天啊
,
2024-10-17 19:31:55
,
所有人可见
,
阅读 1
//扫一下题目,知道是用树实现的,那大概率可以用dfs
//状态表示:f[i][1],f[i][0],第i个节点,选或不选
//属性:max
//状态转移:往下递归
//f[i][1]=f[j][0]
//f[i][0]=f[j][0],f[j][1]
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=6010;
int f[N][2];
int happy[N];
int idx,e[N],ne[N],h[N];
bool k[N];
void add(int a,int b)
{
e[idx]=a,ne[idx]=h[b],h[b]=idx++;
k[a]=1;
}
void dfs(int u)
{
f[u][1]=happy[u];
for(int i=h[u];i!=-1;i=ne[i])
{
int j=e[i];
dfs(j);
f[u][1]+=f[j][0];
f[u][0]+=max(f[j][0],f[j][1]);
}
}
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)cin>>happy[i];
memset(h,-1,sizeof h);
for(int i=1;i<n;i++)
{
int a,b;
cin>>a>>b;
add(a,b);//将a接到b下面
}
int head=0;
for(int i=1;i<=n;i++)
{
if(!k[i])
{
head=i;
break;
}
}
dfs(head);
cout<<max(f[head][0],f[head][1])<<endl;
return 0;
}