P1352 没有上司的舞会

题意：

给你一个树，每个点都有权值，选择一些点使得权值和最大，要求父亲节点和子节点不能同时选择

题解：

经典树形dp
dp[x][0]表示以x为根的子树，且x不参加舞会的最大快乐值
dp[x][1]表示以x为根的子树，且x参加了舞会的最大快乐值
则dp[x][0] = ∑{ max(dp[y][0],dp[y][1]) } (y是x的儿子)
dp[x][1] = ∑{ dp[y][0] } + a[x] (y是x的儿子)
找到唯一的树根root
ans=max(dp[root][0],dp[root][1])

代码：

#include<bits/stdc++.h>
using namespace std;
#define MAXN 6005
int a[MAXN];
int v[MAXN];
vector<int> son[MAXN];
int f[MAXN][2];
void dp(int x)
{f[x][0]=0;f[x][1]=a[x];for(int i=0;i<son[x].size();i++){int y=son[x][i];dp(y);f[x][0]+=max(f[y][0],f[y][1]);f[x][1]+=f[y][0];}
}
int main()
{int n;cin>>n;for(int i=1;i<=n;i++) cin>>a[i];for(int i=1;i<=n-1;i++){int x,y;cin>>x>>y;son[y].push_back(x);v[x]=1;}int root;for(int i=1;i<=n;i++)if(!v[i]) {root=i;break;}dp(root);cout<<max(f[root][0],f[root][1])<<endl;return 0;
}