树形dp 没有上司的舞会

版权声明：未经本蒟蒻同意，请勿转载本蒻博客 https://blog.csdn.net/wddwjlss/article/details/81810252

题意：给定一棵树，每个点有权值，对于每一个点，它和它的父节点不能同时被选，求权值最大值。
定义$dp[x][0]$表示不选择x号点时的最大值，$dp[x][1]$表示选择$x$号点时的最大值。
对于$x$号点不被选择的情况，也就意味着它的子节点$v$可以被选，也可以不被选，所以方程就是：

dp[x][0]+=max(dp[v][1],dp[v][0]);

对于$x$号点被选择的情况，它的子节点$v$不可以被选，所以方程就是

dp[x][1]+=dp[v][0];

我们从根节点开始dfs，从叶节点开始累加，最后在$dp[root][1]和dp[root][0]$中取一个最大值即可

#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;
}