没有上司的舞会

题目不描述，参考大佬代码，具体过程看代码注解:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+6;
int n;
vector<int> g[6010]; //存储孩子
int h[6000][10]; //选择或者不选择当前节点最大嗨皮值
bool yes[maxn];
int a[maxn]; //嗨皮值

void dfs(int x)
{    
    for(int i = 0; i < g[x].size(); i ++) //遍历当前节点的儿子们
    {
        int y = g[x][i]; 
        dfs(y); //去看儿子的儿子
        h[x][1] = h[x][1] + h[y][0]; //选择当前节点，则一定不能选择儿子节点
        h[x][0] = max(h[y][1] + h[x][0],h[y][0]+h[x][0]);
            //不选择当前节点，儿子节点可选择也可不选择，找出最大值
    }
    h[x][1] += a[x];//将选择当前节点的话，加上当前节点的海皮值
}

int main()
{
    scanf("%d",&n);
    for(int i = 1; i <= n; i ++)
    {
        scanf("%d",&a[i]);
    }
    for(int i = 1; i < n; i ++)
    {
        int a, b;
        scanf("%d%d",&a,&b);
        yes[a] = true;        //表示a有父亲
        g[b].push_back(a); //给父亲b保存孩子a
    }
    int a1, b1;
    scanf("%d%d",&a1,&b1);
    for(int i = 1; i <= n; i ++)
    {
        if(!yes[i]) //找到一个没有父亲的 也就是整个图的根节点
        {
            dfs(i);
            printf("%d\n",max(h[i][1],h[i][0]));//根节点选择或者不选择的最大值
            break;
        }
    }

}