没有上司的舞会——详解

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

2. 输入输出格式
   输入格式：
   1.第一行一个整数N。(1<=N<=6000)
   2.接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
   3.接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
   输出格式：输出最大的快乐指数。

3. 输入样例
   7
   1
   1
   1
   1
   1
   1
   1
   1 3
   2 3
   6 4
   7 4
   4 5
   3 5

4. 输出样例
   5

5. 分析
   本题样例的每个点的快乐指数都是1，按照下面的样例可以组成这样的一棵树
   

6. 解题
   设：
   f(i,0/1) 代表以 i 为根的子树的最优解（第二维的值为 0 代表 i 不参加舞会的情况，1 代表 i 参加舞会的情况）
   显然，我们可以推出下面两个状态转移方程（其中下面的 x 都是 i 的儿子）：
   f(i,0)=∑max{f(x,1),f(x,0)}（上司不参加舞会时，下属可以参加，也可以不参加）
   f(i,1)=∑f(x,0)+Ri（上司参加舞会时，下属都不会参加）

以这种方式处理就很好地解决了只判断直接上司的问题了，并且∑的目的是将子树的happy值的和算出来，这也是本题所要求的。

7. 代码如下

#include <iostream>
#include<vector>
using namespace std;

const int N = 1e4 + 10;
vector<int> tr[N];          					//tr[N]数组用来存子树
int f[N][2], v[N], Happy[N], n;
void dfs(int u)
{
    f[u][0] = 0;
    f[u][1] = Happy[u];
    for (auto v : tr[u])                        //取tr[u]里面的值         v是u的下属
    {
        dfs(v);
        f[u][0] += max(f[v][0], f[v][1]);       //如果上司不来，直接下属可来可不来
        f[u][1] += f[v][0];                     //如果上司来，直接下属肯定不来
    }
}
int main() {
    cin.tie(nullptr)->sync_with_stdio(false);   //cin，cout之所以效率低，是因为先把要输出的东西存入缓冲区，再输出，导致效率降低，而这段语句可以来打消iostream的输入 输出缓存，可以节省许多时间，使效率与scanf与printf相差无几
    cin >> n;                                   //n个职员
    for (int i = 1; i <= n; ++i)
        cin >> Happy[i];                        //每个职员来的快乐指数存成一个数组
    for (int i = 1, x, y; i < n; ++i)
    {
        cin >> x >> y;                          //y是x的直接上司
        v[x] = 1;								//把不是BOSS的上司的v值全部变为1
        tr[y].push_back(x);						//建树
    }
    int root;
    for (int i = 1; i <= n; ++i)                //找boss
    {
        if (!v[i])
        {
            root = i;							//把当前i设置为根节点也就是BOSS
            break; 
        }
    }
    dfs(root);
    cout << max(f[root][0], f[root][1]) << "\n";//比较BOSS不来和BOSS来的最大快乐值
}

8. 解析
   其实上面代码已经标注的很明确了，如果理解还是有困难，可以看一看下面的过程，希望可以对你有帮助。
   输入快乐指数的部分没有什么可说的，从输入上下级关系开始看

输入	v[i]	tr[u]
1 3	v[1] = 1	tr[3] = 1
2 3	v[2] = 1	tr[3] = 2
6 4	v[6] = 1	tr[4] = 6
7 4	v[7] = 1	tr[4] = 7
4 5	v[4] = 1	tr[5] = 4
3 5	v[3] = 1	tr[5] = 3

从上面可以画出完整的二叉树结构，然后通过 DFS，在返回上一层时更新当前结点的最优解。