【洛谷 P1352】【codevs1380】没有上司的舞会

链接：https://www.luogu.org/problem/show?pid=1352
题意：求n个点的最大带权独立集
算法：显然树形DP
* 分析：用dp[u][1]表示选第u个人时以u为根的子树的最大欢乐度，dp[u][0]表示不选第u个人时以u为根的子树的最大欢乐度。
* 又上司不能和下属同选，则有
* dp[u][1]=sigma(dp[v][0])+w[i]{v是u的儿子结点}
* dp[u][0]=sigma(max(dp[v][0],dp[v][1])){v是u的子结点}
* 边界条件:如果i是叶子，则dp[u][0]=0;dp[u][1]=w[i];

第一道树型DP，感觉要主要考虑
1.更新的时候子节点和祖先节点的关系
2.储存数据的方式

下附代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int f[10000][2],happy[10000],fa[10000];
int n;
void dfs(int x)
{
    for(int i=1;i<=n;i++)
    if(fa[i]==x)
    {
        dfs(i);
        f[x][0]+=max(f[i][0],f[i][1]);
        f[x][1]+=f[i][0];
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&f[i][1]);
    int aa,bb;
    scanf("%d %d",&aa,&bb); 
    while(aa&&bb) 
    {
        fa[aa]=bb;
        scanf("%d%d",&aa,&bb);
    }
    int root=1;
    while(fa[root]!=0) root=fa[root];
    dfs(root);
    printf("%d",max(f[root][0],f[root][1]));
}