洛谷 1352 没有上司的舞会 树形DP

题目：
https://www.luogu.org/problem/show?pid=1352；

套路DP；
由于点权可能为负；
那么上司不来时，下属可来，可不来；
但上司来时，下属一定不能来；

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

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int MAXN=20001;
int dp[MAXN][4],fst[MAXN],nxt[MAXN],num[MAXN];
int tot,n,fa[MAXN],ans;
struct hh
{
    int from,to;
}ma[MAXN];
void build(int f,int t)
{
    tot++;
    ma[tot]=(hh){f,t};
    nxt[tot]=fst[f];
    fst[f]=tot;
    return;
}
void dfs(int x)
{
    for(int i=fst[x];i;i=nxt[i])
    {
        int v=ma[i].to;
        if(fa[x]==v) continue;
        dfs(v);
        dp[x][0]+=max(dp[v][0],dp[v][1]);
        dp[x][1]+=dp[v][0];
    }
    return;
}
void solve()
{
    int x,y;
    cin>>n;
    for(int i=1;i<=n;i++) scanf("%d",&num[i]),dp[i][1]=num[i];
    while( ~scanf("%d%d",&x,&y))
    {
        if(x==0 && y==0) break;
        fa[x]=y;
        build(x,y);
        build(y,x);
    }
    for(int i=1;i<=n;i++)
        if(fa[i]==0) {dfs(i);}
    for(int i=1;i<=n;i++) ans=max(ans,max(dp[i][0],dp[i][1]));
    cout<<ans;
}
int main()
{
    solve();
    return 0;
}