洛谷 1352 没有上司的舞会

** 原题：**
题目描述

某大学有 nnn 个职员，编号为 1…n1\ldots n1…n。

他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。

现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数 rir_iri​，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。

所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。
输入格式

输入的第一行是一个整数 nnn。

第 222 到第 (n+1)(n + 1)(n+1) 行，每行一个整数，第 (i+1)(i+1)(i+1) 行的整数表示 iii 号职员的快乐指数 rir_iri​。

第 (n+2)(n + 2)(n+2) 到第 2n2n2n 行，每行输入一对整数 l,kl, kl,k，代表 kkk 是 lll 的直接上司。
输出格式

输出一行一个整数代表最大的快乐指数。
输入输出样例

输入：
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5

输出：
5
说明/提示
数据规模与约定

对于 100%100%100% 的数据，保证 1≤n≤6×1031\leq n \leq 6 \times 10^31≤n≤6×103，−128≤ri≤127-128 \leq r_i\leq 127−128≤ri​≤127，1≤l,k≤n1 \leq l, k \leq n1≤l,k≤n，且给出的关系一定是一棵树。

代码：

#include <bits/stdc++.h>

using namespace std;
const int maxn = 6e3 + 5;

int n;
int r[maxn];
vector<int> t[maxn];
int mark[maxn];
bool vis[maxn];
int dp[maxn][2];
void dfs(int root) {
    if (vis[root]) {
        return;
    }
    vis[root] = true;

    for (auto child : t[root]) {
        if (!vis[child]) {
            dfs(child);
            dp[root][0] += max(dp[child][0], dp[child][1]);
            dp[root][1] += dp[child][0];
        }
    }
}

int main()
{
    ios::sync_with_stdio(false);
    while(cin >>n ) {
        for (int i=1;i<=n;i++) {
            cin >> dp[i][1];
        }
        int l,k;
        memset(mark,0,sizeof(mark));
        memset(vis, false, sizeof(vis));
        for (int i=1;i<n;i++) {
            cin >> l >> k;
            t[k].push_back(l);
            mark[l] = 1;
        }
        for (int i=1;i<=n;i++) {
            if (mark[i] == 0) {
                dfs(i);
                cout << max(dp[i][0], dp[i][1]) << endl;
                break;
            }
        }

    }
	return 0;
}

解答：

树型dp入门的题目
设置$dp[i][0/1]$ ，其中$dp[i][0]$表示当前节点i不参加时，以i为节点的子树所得到的最大分数;$dp[i][1]$表示节点i参加时以i为节点的子树所得到的最大分数。

那么，当节点i不参加时，其子节点可以选择参加也可以不参加，取最大值，有
$dp[i][0]+=max(dp[child[i]][0]],dp[child[i]][1])$
当节点i参加时，其子节点不参加
$dp[i][1]+=max(dp[child[i]][0])$