LuoGu P1352 没有上司的舞会

题目传送门
这可能是最简单的树形Dp了吧
对于每个人，要么他来，他的下属不来
要么他不来，他的下属爱来不来
于是设计状态：
f[i][0/1]表示以i为根的子树中最大能达到的快乐值(i这个人选或者不选)
然后一遍dfs一遍转移就好了

#include <iostream>
#include <cstdlib>
#include <cstdio>
#define max(a,b) (a>b?a:b)

using namespace std;

const int N=6e3+5;

struct edge{
    int to,next,pre;
}e[N];

int f[N][2],n,val[N],u,v,head[N],tot,ans=0;
bool had[N];

inline void build(int u,int v){
    e[++tot].next=head[u];e[tot].pre=u;
    head[u]=tot;e[tot].to=v;
    return ;
}

inline void work(int x){
    f[x][0]=0;f[x][1]=val[x];
    for(int i=head[x];i;i=e[i].next){
        int k=e[i].to;
        work(k);
        f[x][0]+=max(f[k][1],f[k][0]);
        f[x][1]+=f[k][0];
    }
    return ;
}

int main(){
    scanf("%d",&n);
    for(register int i=1;i<=n;++i) scanf("%d",&val[i]);
    do{
        scanf("%d%d",&u,&v);
        build(v,u);had[u]=true;
    }while(u!=0&&v!=0);
    for(int i=1;i<=n;++i) if(!had[i]) work(i);
    for(register int i=1;i<=n;++i) ans=max(ans,max(f[i][1],f[i][0]));
    printf("%d\n",ans);
    return 0;
}