P1352 没有上司的舞会 (树形dp)

题意：给你一颗树，每个节点都有一个权值，如果你选择了他的根节点，你就没有办法选择这个节点的儿子节点，现在问你怎样选择可以使得选的的权值最大。

思路：首先dp[i][j] 分别表示的是，你在以i为根节点的子树中，选择了根节点dp[i][0] , 或者选择了根节点dp[i][1]，详细情况看代码把

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 6000 + 10;
vector<int>G[maxn];
int a[maxn] , du[maxn] , dv[maxn] , dp[maxn][2];
void dfs(int u)
{
    dp[u][0] = 0; // 没有选这个点
    dp[u][1] = a[u]; // 选了这个点
    for(int i = 0 ; i < G[u].size() ; i++)
    {
        int v = G[u][i];
        dfs(v);
        dp[u][0] += max(dp[v][0] , dp[v][1]); // 如果没有选择这个点，那么下一个节点我们会有两种选择 1 选择他的子节点，2 不选择他的子节点 
        dp[u][1] += dp[v][0]; // 如果选了这个点，那么就只能选择不选他的子节点 
    }
}
int main()
{
    int n , u , v , root;
    scanf("%d",&n);
    for(int i = 1 ; i <= n ; i++) scanf("%d",&a[i]);
    for(int i = 0 ; i < n -1 ; i++)
    {
        scanf("%d%d",&u,&v);
        G[v].push_back(u);
        dv[u] ++;
    }
    int a,b;
    scanf("%d%d",&a,&b);
    for(int i = 1 ; i <= n ; i++)
    {
        if(dv[i] == 0) {root = i ; break;}
    }
    dfs(root);
    printf("%d\n",max(dp[root][0] , dp[root][1]));

}