动态规划（算法竞赛、蓝桥杯）--树形DP没有上司的舞会

1、B站视频链接：E17 树形DP Luogu P1352 没有上司的舞会_哔哩哔哩_bilibili

题目链接：没有上司的舞会 - 洛谷

#include <bits/stdc++.h>
using namespace std;
const int N=6010;
int n;
int w[N];
vector<int> a[N];
bool fa[N];
int f[N][2];

void dfs(int u){
	f[u][1]=w[u];//选根节点的快乐指数先加上
	for(int v:a[u]){//遍历u的子节点,选或者不选
		dfs(v);//深搜到底后返回
		f[u][0]+=max(f[v][0],f[v][1]);
		f[u][1]+=f[v][0]; 
	}
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)cin>>w[i];
	for(int i=0;i<n-1;i++){
		int x,y;
		cin>>x>>y;//y是x的直接上司 
		a[y].push_back(x);
		fa[x]=true;//下属都有父节点 
	}
	int root=1;
	while(fa[root])root++;//找到没有上司的即为根节点 
	dfs(root);
	cout<<max(f[root][0],f[root][1]);

	return 0;
}