hdu 1520 Anniversary party (树形dp)

题目链接

题目描述：

给定 n 个人，没人一个欢乐值，参加派对的总欢乐值等于所有参与的人的欢乐值之和，接下来输入多组数，a,b表示b 是 a 的上司，没有人想和自己的直系上司共同参与派对，求最大欢乐值

树形dp入门题目，因为我们搜索的顺序实在一颗树上进行的，因此可以直接采用递归的方法进行求解。

首先用一个邻接表或者邻接矩阵存放树上的信息，将没有父节点的节点设为根节点，从根节点开始搜索。

搜索过程中，对于每一个点，有两个状态，分别是参与派对和不参与派对。

用一个二维数组dp[maxn][2]保存这个状态，对于每一个dp[i][j]表示第 i 个人是否参与派对的最大欢乐值，如果第 i 个人参与，那么他的上司就不能参与（因为我们的更新顺序是从根到子节点，也就是说先遇到的都是后面的上司），所以可以得到方程

dp[x][1] += dp[xx][0];

如果不参与则有两种情况，他的上司参与或者不参与，取最大值即可

dp[x][0] += max(dp[xx][0], dp[xx][1]);

在每一次进入新的递归中，要将dp[i][0] 和 dp[i][1]初始化，dp[i][0]的值为0，dp[i][1]为第 i 个人的欢乐值，先递归在进行计算，由后往前推即可。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <vector>
using namespace std;
const int maxn = 6010;
int n, L ,K, m;
int a[maxn], v[maxn];
int dp[maxn][2];// 以 i 为根子树，是否包含根位置的解 
vector<int> son[maxn];
int dfs(int x){
	dp[x][0] = 0;// 不包含x 
	dp[x][1] = a[x];// 包含x 
	int len = son[x].size();
	for (int i = 0; i < len; i++){//遍历儿子节点 
		int xx = son[x][i];
		dfs(xx);
		dp[x][1] += dp[xx][0];
		dp[x][0] += max(dp[xx][0], dp[xx][1]);
	}
}
int main()
{
	while(~scanf("%d", &n)){
		memset(v, 0, sizeof v);
		memset(a, 0, sizeof a);
		for (int i = 1; i <= maxn; i++)son[i].clear();
		for (int i = 1; i <= n; i++)scanf("%d", &a[i]);
		while (scanf("%d %d", &L, &K) && L && K){
			son[K].push_back(L);
			v[L] = 1;
		}
		int root;
		for (int i = 1; i <= n; i++){
			if(v[i] == 0){
				root = i;
				break;
			}
		}
		dfs(root);
		printf("%d\n", max(dp[root][0], dp[root][1]));
	}
	return 0;
}