洛谷 P1352 没有上司的舞会

题目链接

【题目描述】

某大学有 nnn 个职员，编号为 1…n。
他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。
现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数 $r_i$，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。
所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

【输入】

输入的第一行是一个整数 n。
第 2 到第 (n+1) 行，每行一个整数，第 (i+1) 行的整数表示 i 号职员的快乐指数 $r_i$。
第 (n+2) 到第 2n 行，每行输入一对整数 l,k，代表 k 是 l 的直接上司。

【输出】

输出一行一个整数代表最大的快乐指数

【源代码】

#include <bits/stdc++.h>
using namespace std;
const int N = 10010;
vector<int>s[N];
int f[N][2];//f[i][0]是i不参加的最大值，f[i][1]是参加的
//f[x][0] = ∑{max{f[y][0],f[y][1]}}
//f[x][1] = ∑{f[y][0] + happy[i]}
//y是x的下属
int h[N];//开心值
bool vis[N];
void treeDP(int x){
    f[x][0] = 0;
    f[x][1] = h[x];
    for(int i=0;i<s[x].size();i++){
        int y = s[x][i];
        treeDP(y);
        f[x][0] += max(f[y][0],f[y][1]);
        f[x][1] += f[y][0];
    }
}
int main()
{
    int n;
    cin >> n;
    for(int i=1;i<=n;i++)
        cin >> h[i];
    for(int i=1;i<n;i++){
        int x,y;
        cin >> x >> y;
        s[y].push_back(x);
        vis[x] = 1;
    }
    int root;
    for(int i=1;i<=n;i++){
        if(!vis[i]){
            root = i;
            break;
        }
    }
    treeDP(root);
    int res = max(f[root][0],f[root][1]);
    cout << res <<endl;
    return 0;
}