洛谷 P1352 没有上司的舞会

P1352 没有上司的舞会

树形dp

dp[i][0]表示第i个人没有来的活跃值

dp[i][1]表示第i个人来了的活跃值

第i个人来时，他的直接下属就不可以来  所以  如果k是他的直接下属的话  dp[i][1]+=dp[k][0]

第i个人不来，他的直接下属可以来也可以不来 所以  如果k是他的直接下属的话  dp[i][0]+=max(dp[k][0],dp[k][1])

#include<bits/stdc++.h>
using namespace std;
#define maxn 1000000
int n,m,fa[maxn],root,dp[maxn][2],x,y;
bool vis[maxn];

void tree_dp(int x)
{
    vis[x]=1;
    for(int i=1;i<=n;i++)
    {
        if(!vis[i]&&fa[i]==x)
        {
            tree_dp(i);
            dp[x][1]+=dp[i][0];
            dp[x][0]+=max(dp[i][1],dp[i][0]);
        }
        
    }
}

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&dp[i][1]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        fa[x]=y; root=x;
    }
    scanf("%d%d",&x,&y);
    while(fa[root]) root=fa[root];
    tree_dp(root);
    printf("%d\n",max(dp[root][0],dp[root][1]));
    return 0;
}