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;//x是y的直接上司a[y].push_back(x);fa[x]=true; //x有父节点 }int root=1;while(fa[root])root++;//找到根节点dfs(root);cout<<max(f[root][0],f[root][1]);return 0;
}