P1352 没有上司的舞会
一道树形DP的模板题。树形DP一般用递归实现,将节点从小到大的顺序作为动态规划的阶段,先递归子节点,再回溯从子节点到另一节点转移。
#include<bits/stdc++.h>
using namespace std;
vector son[1010010];
int dp[10001][101],vis[100001],tree[100101];
int n;
void dptree(int x){
dp[x][0] = 0;
dp[x][1] = tree[x];
for(int i=0;i<son[x].size();i++){
int Tree = son[x][i];
dptree(Tree);
dp[x][0]+=max(dp[Tree][0],dp[Tree][1]);
dp[x][1]+=dp[Tree][0];
}
}
int main(){
cin >> n;
for(int i=1;i<=n;i++){
cin >> tree[i];
}
for(int i=1;i<n;i++){
int x,y;
cin >> x >>y;
vis[x] = 1;
son[y].push_back(x);
}
int root;
for(int i=1;i<=n;i++){
if(!vis[i]){
root = i;
}
}
dptree(root);
int ans = max(dp[root][0],dp[root][1]);
cout << ans;
return 0;
}```