AcWing 285. 没有上司的舞会

AcWing 285. 没有上司的舞会

Ural大学有N名职员，编号为1~N。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 Hi 给出，其中 1≤i≤N。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式
第一行一个整数N。

接下来N行，第 i 行表示 i 号职员的快乐指数Hi。

接下来N-1行，每行输入一对整数L, K,表示K是L的直接上司。

输出格式
输出最大的快乐指数。

数据范围
1≤N≤6000,
−128≤Hi≤127
输入样例：

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5

输出样例：

5

树形dp在我印象里，应该要用递归实现，因为子节点需要遍历。
所以用递归实现dp，我们还是闫氏分析，找状态表示和状态计算。
状态表示

$dp[i][j]$ ，$j$为1的时候表示以i为根节点，选i。 $j$不为1的时候表示以i为根节点，不选i

然后求的是最大值。

状态计算
$dp[i][0]=\sum max(dp[j][1],dp[j][0])j是i的子节点$
$dp[i][1]=w[i]+\sum dp[j][0]j是i的子节点$

代码如下

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;

const int N=6010;

int ne[N],h[N],e[N],idx;
int w[N];
int dp[N][N];
bool get_father[N];

void add(int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}

void dfs(int x)
{
    dp[x][1]=w[x];
    for(int i=h[x];i!=-1;i=ne[i])
    {
        int j=e[i];
        dfs(j);
        dp[x][0]+=max(dp[j][1],dp[j][0]);
        dp[x][1]+=dp[j][0];
    }
    return;
}


int main(void)
{
    memset(h,-1,sizeof h);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>w[i];
    for(int i=1;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        get_father[a]=true;
        add(b,a);
    }
    int root=1;
    while(get_father[root]) root++;
    dfs(root);
    int res=max(dp[root][0],dp[root][1]);
    cout<<res;
}