题意:给一棵树每个节点给定一个价值,子节点和父节点不能同时选取,求能够选取的最大价值。
用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;
}