没有上司的舞会【树形DP】

题目描述

【洛谷】没有上司的舞会

某大学有$n$个职员，编号为$1...n$。

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

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

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

输入格式

输入的第一行是一个整数$n$。

第$2$到第$n+1$行，每行一个整数，第$i+1$行的整数表示$i$​号职员的快乐指数$r_i$。

第$n+2$到第$2n$行，每行输入一对整数$l,k$，代表$k$是$l$​的直接上司。

输出格式

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

输入输出样例

输入 #1

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5

输出 #1

5

说明/提示

数据规模与约定

对于$100\%$​的数据，保证$1\le n\le 6\times 10^3,-128\le r_i\le 127,1\le l,k\le n$​且给出的关系一定是一棵树。

解题思路及代码

我们可以定义$f(i,0/1)$代表$i$为根的子树的最优解，也就是$f(i,0)$代表$i$不参加舞会的最大快乐指数，$f(i,1)$代表$i$参加舞会的最大快乐指数。

那么就可以推导出如下的状态转移方程(下面的$x$代表$i$的职员)：

• f ( i , 0 ) = ∑ m a x { f ( x , 0 ) , f ( x , 1 ) } f(i,0)=\sum{max\{f(x,0),f(x,1)\}} f(i,0)=∑max{f(x,0),f(x,1)}​​

  上司不参加舞会的时候，下属可以参加，也可以不参加，因此取两者的最大值。但是一个上司下面有若干个职员，因此需要求和。

• $f(i,1)=\sum f(x,0)+a_i$

  上司参加舞会的时候，所有的职员都不参加舞会。

AC代码：

#include <bits/stdc++.h>
using namespace std;

vector<int> tree[6010];//存储树
int happy[6010];//happy[i]代表i的快乐指数
int f[6010][2];//f[i][0]:i不参加时的最大快乐指数，i参加时的最大快乐指数。
bool isVisited[6010];//是否已经访问


void dfs(int index){
    f[index][0] = 0;
    f[index][1] = happy[index];
    for(int i=0; i<tree[index].size();i++){
        int next = tree[index][i];
        if(!isVisited[next]){
            isVisited[next] = true;
            dfs(next);
            f[index][0] += max(f[next][0], f[next][1]);
            f[index][1] += f[next][0];
        }
    }
}

int main(){
    int n;
    cin>>n;
    for(int i=1; i<=n; i++){
        cin>>happy[i];
    }
    int l,k;
    for(int i=0;i<n-1;i++){
        cin>>l>>k;
        tree[l].push_back(k);
        tree[k].push_back(l);
    }

    isVisited[1]=true;
    dfs(1);
    //求最大的快乐指数
    int maxn = f[1][0];
    for(int i=1; i<=n; i++){
        if(f[i][0]>maxn) maxn = f[i][0];
        if(f[i][1]>maxn) maxn = f[i][1];
    }
    cout<<maxn<<endl;
    return 0;
}