本题是经典的树型dp问题。
原题如下:
![](https://i-blog.csdnimg.cn/blog_migrate/73146e3decde87adc124c81977685eac.png)
状态转移方程推导如下:
将状态分为0,1两部分,0为不选择此位,1为选择此位
f[N][2] 前一位代表哪一位,后一位表示是否选择
当选择该位时,其下属则无法参加舞会,
即f[N][1] = sum(f[N的下属][0])
当不选择该位时,其下属则可以选择是否参加舞会,
即f[N][0] = sum(max(f[N的下属][0],f[N的下属][1]))
即可得状态转移方程。
代码如下:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 6010;
int f[N][2];
int e[N], ne[N], h[N], idx;
int happy[N];
bool have_father[N];
void add(int a, int b) // 添加一条边a->b
{
have_father[b] = true;
e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs(int x)
{
f[x][1] = happy[x];
for(int i = h[x]; i != -1; i = ne[i])
{
int j = e[i];
dfs(j);
f[x][1] += f[j][0];
f[x][0] += max(f[j][0], f[j][1]);
}
}
int main()
{
int n;
cin >> n;
for (int i = 1; i <= n; i ++ )
{
cin >> happy[i];
}
memset(h, -1, sizeof h);
for (int i = 1; i < n; i ++ )
{
int a, b;
cin >> b >> a;
add(a, b);
}
int root = 1;
while(have_father[root])
{
root ++;
}
dfs(root);
cout << max(f[root][0], f[root][1]) << endl;
}