Anniversary party（树形dp）

HDU - 1520 

题目描述

要有一个庆祝乌拉尔州立大学80周年。这所大学有一个员工的层次结构。这意味着主管的关系形成一个树在特列季亚科夫校长”。为了使党的滑稽的每一个，校长不希望员工和他或她的直属主管到场。人事办公室评估每个员工的欢乐，所以每个人都有一些数（等级）连接到他或她。你的任务是与客人欢宴评级的最大金额列出客人。

输入

员工编号从1到N的第一行输入包含一个数N = N = 6 < 1 < 000。接下来的N行包含相应员工的欢乐度。欢乐的评级是一个整数的范围从128到127 -。之后去行，描述一个主管关系树。每行树规范的形式：

L K

这意味着，第k个员工是l个员工的直接主管。输入用行结束。

0 0

输出

输出应包含顾客评分之和最大。

样本输入

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

示例输出

5

 

题意：有n个人，接下来n行是n个人的权值，再接下来n行给出L，K，表示K是L的上司，L和K不能同时出现。

思路：

用dp数据来记录价值，价值为活跃度

状态转移方程为：

DP[i][1]+=DP[j][0];//上司被选中，下属不选

DP[i][0]+=max{DP[j][0],DP[j][1]};//上司不被选中，下属可选可不选

从根节点r开始进行dfs，得到最后的结果为max{[DP[r][0],DP[r][1]}。

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e4+10;
int dp[MAXN][2];
int fa[MAXN];
vector<int>G[MAXN];
int n;
void dfs(int u) {
    int v;
    for(int i = 0; i < G[u].size(); i++) {
        v = G[u][i];
        dfs(v);
        dp[u][1] += dp[v][0];
        dp[u][0] += max(dp[v][1], dp[v][0]);
    }
}
int main(){
	int L,K,root;
	while(cin>>n){
		memset(dp,0,sizeof(dp));
		memset(fa,0,sizeof(fa));
		for(int i = 1;i <= n; i++){
			cin>>dp[i][1];
			G[i].clear();
		}
		while(cin>>L>>K && (L+K)){
			fa[L]++;
			G[K].push_back(L);
		}
		for(int i=1;i<=n;i++){
			if(fa[i]==0){
				root=i;
				break;
			}
		}
		dfs(root);
		printf("%d\n",max(dp[root][1],dp[root][0]));
	}
	
	return 0;
}