285. 没有上司的舞会

#include<bits/stdc++.h>

using namespace std;

const int N =6010;

int n;
int h[N],e[N],ne[N],idx;
int happy[N];
int f[N][2];
bool has_fa[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;i=ne[i])
    {
        int j=e[i];
        dfs(j);
        
        f[u][1]+=f[j][0];
        f[u][0]+=max(f[j][0],f[j][1]);
    }
}

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;
        add(b,a);
        has_fa[a]=true;
    }
    
    int root=1;
    while(has_fa[root]) root++;
    
    dfs(root);
    
    cout<<max(f[root][0],f[root][1])<<endl;
    
    return 0;
}

f[i][1]表示选该节点

f[i][0]表示不选该节点

建立邻接表差不多忘光了

如果是选择该点，就不能选上司，就是状态转移方程的意思

不选该点，选不选上司取一个最大值（最优解）即可

最后看是否要选校长（？），因为只需要一部分职员参加这个舞会

还是比较容易理解