洛谷 题解 P1352 【没有上司的舞会】

树形DP的一道较为基础的模板题

• 状态

dp[i][0/1]为第i个员工是否来参加的最大值

• 转移

先找到根节点
先遍历完它的儿子，再来更新答案
dp[i][0]+=max(dp[j][0],dp[j][1]);//j为i的儿子，i不来，那j可来可不来
dp[i][1]+=dp[j][0];//j为i的儿子，i来，那只能不来

• 初始

dp[i][0]=0;//不来的初始值为0
dp[i][1]=r[i];//来的初始值就是此人的快乐指数

• 答案

很明显就是大BOSS来不来的问题了
max(dp[root][0],dp[root][1]);

完整代码:

#include<bits/stdc++.h>
using namespace std;
const int N=6000+10;
int n,m;
int r[N];
bool v[N];
vector<int>son[N];
int dp[N][2];
inline int read()
{
    int tot=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9')
    {
        if(c=='-')f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9')
    {
        tot=tot*10+c-'0';
        c=getchar();
    }
    return tot*f;
}
inline void f(int now)
{
    dp[now][0]=0;
    dp[now][1]=r[now];
    for(int i=0;i<son[now].size();i++)
    {
        int x=son[now][i];
        f(x);
        dp[now][0]+=max(dp[x][0],dp[x][1]);
        dp[now][1]+=dp[x][0];
    }
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)r[i]=read();
    int x,y;
    for(int i=1;i<n;i++)
    {
        x=read();y=read();
        son[y].push_back(x);
        v[x]=1;
    }
    int root;
    for(int i=1;i<=n;i++)
    if(!v[i])
    {
        root=i;
        break;  
    }
    f(root);
    int ans=max(dp[root][1],dp[root][0]);
    cout<<ans<<endl;
    return 0;
}