版权声明:未经本蒟蒻同意,请勿转载本蒻博客 https://blog.csdn.net/wddwjlss/article/details/81810252
题意:给定一棵树,每个点有权值,对于每一个点,它和它的父节点不能同时被选,求权值最大值。
定义
表示不选择x号点时的最大值,
表示选择
号点时的最大值。
对于
号点不被选择的情况,也就意味着它的子节点
可以被选,也可以不被选,所以方程就是:
dp[x][0]+=max(dp[v][1],dp[v][0]);
对于 号点被选择的情况,它的子节点 不可以被选,所以方程就是
dp[x][1]+=dp[v][0];
我们从根节点开始dfs,从叶节点开始累加,最后在 中取一个最大值即可
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int dp[1001000][2],f[1001000];
int n,num,head[1001000];
struct node
{
int next,to;
}e[1001000];
void add(int from,int to)
{
e[++num].next=head[from];
e[num].to=to;
head[from]=num;
}
void dfs(int x)
{
for(int i=head[x];i!=0;i=e[i].next)
{
int v=e[i].to;
dfs(v);
dp[x][0]+=max(dp[v][1],dp[v][0]);
dp[x][1]+=dp[v][0];
}
return;
}
int main()
{
cin>>n;
for(int i=1;i<=n;++i)
scanf("%d",&dp[i][1]);
for(int i=1;i<=n-1;++i)
{
int x,y;
scanf("%d%d",&x,&y);
add(y,x);
f[x]=y;
}
int z,ze;
cin>>z>>ze;
int root;
for(int i=1;i<=n;++i)
if(f[i]==0)
{
root=i;
break;
}
dfs(root);
int ans=max(dp[root][1],dp[root][0]);
cout<<ans;
return 0;
}