P1352 没有上司的舞会

--------------------

链接：Miku

--------------------

很入门的树形dp，首先，在这个题中，我们要做的就是求出来每一个子节点，然后用他们去更新父亲节点。

对于每一个节点，他有两种状态，去，或者不去，我们定义dp[i][0]为第i个节点也去的状态，而dp[i][1]为它不去，那么很显然

如果这个点去了，它的子节点肯定不去，那么dp[i][1]=dp[孩子们][0]的和，

如果这个点不去，它的子节点可以去也可以不去，那么就是dp[i][0]=max(dp[孩子们][0],dp[sons][1])的和

----------------------

#include<iostream>
#include<map>
#include<cstdio>
using namespace std;
int n;
int r[6001];
int ru[6001];
int x,y;
int p;
int dp[6001][2];
int head[6001];
struct b{
    int t;
    int ne;
} e[6001];
void add(int f,int to){
    p++;
    e[p].t=to;
    e[p].ne=head[f];
    head[f]=p;
}
int ro;
void dfs(int now){
    dp[now][0]=0;
    dp[now][1]=r[now];
    for(int i=head[now];i;i=e[i].ne){
        dfs(e[i].t);
        dp[now][0]+=max(dp[e[i].t][1],dp[e[i].t][0]);
        dp[now][1]+=dp[e[i].t][0];
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;++i){
        scanf("%d",&r[i]);
    }
    for(int i=1;i<=n-1;++i){
        scanf("%d%d",&x,&y);
        add(y,x);
        ru[x]++;
    }
    for(int i=1;i<=n;++i){
//        cout<<ru[i]<<" ";
        if(!ru[i]){
            ro=i;
            break;
        }
    }
//    cout<<ro<<endl;
    dfs(ro);
    cout<<max(dp[ro][0],dp[ro][1]);
    return 0;
}