洛谷P1352 没有上司的舞会 //经典树形dp

题意：给一棵树每个节点给定一个价值，子节点和父节点不能同时选取，求能够选取的最大价值。

用dp[u][0]表示不选取u节点的u子树（包括u在内）能获得的最大价值，dp[u][1]表示选取u节点的u子树（包括u在内）能获得的最大价值。

#include<bits/stdc++.h>
using namespace std;
int dp[10001][2],r[10010],vis[10010];
vector<int>q[10010];
void dfs(int u){
    dp[u][0]=0;dp[u][1]=r[u];
    for(int i=0;i<q[u].size();i++){
        int v=q[u][i];
        dfs(v);
        dp[u][0]+=max(dp[v][0],dp[v][1]);
        dp[u][1]+=dp[v][0];
    }
}
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>r[i];
    }
    for(int i=1;i<=n-1;i++){
        int u,v;
        cin>>u>>v;
        q[v].push_back(u);
        vis[u]=1;
    }
    int a,b;cin>>a>>b;
    int root;
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            root=i;break;
        }
    }
//    cout<<root<<endl;
    dfs(root);
    cout<<max(dp[root][0],dp[root][1])<<endl;
}