AcWing 285. 没有上司的舞会
原题链接
简单
作者:
我要出去乱说
,
2021-01-30 21:33:02
,
所有人可见
,
阅读 261
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 6010;
int n;
int happy[N];
int h[N], e[N], ne[N], idx;
int f[N][2]; //f[i][1]表示以i为根的子树中选择的方案,并且包括i节点
bool has_father[N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs(int u)
{
f[u][1] = happy[u];
for (int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
dfs(j);
f[u][0] += max(f[j][0], f[j][1]); //注意此处是根节点累加子节点
f[u][1] += f[j][0];
}
}
int main()
{
cin >> n;
for (int i = 1; i <= n; i ++ ) cin >> happy[i];
memset(h, -1, sizeof h);
for (int i = 0; i < n - 1; i ++ )
{
int a, b;
cin >> a >> b; //b是a的父节点
has_father[a] = true;
add(b, a);
}
int root = 1;
while (has_father[root]) root ++ ; //循环遍历找出根节点,即没有父节点的节点
dfs(root);
cout << max(f[root][0], f[root][1]);
return 0;
}