AcWing 285. 没有上司的舞会
原题链接
简单
作者:
rushhhhh
,
2021-02-28 00:28:46
,
所有人可见
,
阅读 433
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
/*
状态表示f[i,0], f[i,1]
集合:(用0、1表示)邀请/不邀请i号职员参加舞会
属性:max
状态计算
不选i,则f[i,0] += max(f[s,0], f[s,1])
其中,遍历所有子结点。子结点当然可选可不选,故取其max
选i,则f[i,1] += f[s,0]
同样遍历所有子结点。子结点只能不选(因为父结点i选了)
*/
const int N = 6010;
int n;
int happy[N];
int h[N], e[N], ne[N], idx;
int f[N][2];
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];
// 先递归到底,自底往上算出f
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(f));
for(int i=0; i < n-1; i++)
{
int a, b;
cin >> a >> b;
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]);
}