HDU 1520 Anniversary party (树形dp) 解题报告

个人感觉这是一道比较经典而且基础的树形dp入门题目。

树形dp本质上就是在树的结构上去动态决策。具体到这道题上，可以这样说：对于某个人，有两种决策，让他来或者不让他来，如果让他来，那么他的直接下级肯定不能来；如果不让他来，那么它的直接下级来不来都行。

那么我们就可以得出解法了：首先，从输入中找出根节点（即最高级的上司），从他开始深搜。搜索的内容是以该节点为根节点的子树所能获得的最大开心值。

设dp[i][0]为第i个人不来时以i为根节点的子树的最大开心值，dp[i][1]为第i个人来时的值。j为i的子节点

dp[i][0]=∑max(dp[j][0],dp[j][1])           //第i个人不来时，下级来不来都可以，选开心值大的

dp[i][1]=∑(dp[j][0])    //第i个人来时，他的下级不能来，只能去dp[j][0]

         对于具体的代码实现方面，我的习惯是：

（1）    用vector存储某个结点的所有子节点

（2）    在输入时，先设第一个结点为根节点，之后如果某对输入中，子节点为目前的根节点，就把这对输入的根节点设为总的根节点。具体见代码

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int MAXN = 6005;
int dp[MAXN][2];
int happy[MAXN];
vector<int>child[MAXN];
void dfs(int rt)
{
	if (child[rt].size() == 0)
	{
		dp[rt][0] = 0;
		dp[rt][1] = happy[rt];
		return;
	}
	int i;
	dp[rt][1] = happy[rt];		//要把自己算上
	for (i = 0; i < child[rt].size(); ++i)
	{
		int next = child[rt][i];
		dfs(next);
		dp[rt][0] += max(dp[next][0],dp[next][1]);
		dp[rt][1] += dp[next][0];
	}
}
void ini(int n)
{
	memset(dp,0,sizeof(dp));
	int i;
	for (i = 1; i <= n; ++i)
		child[i].clear();
}
int main()
{
	int n,i,l,k,rt;
	while (scanf("%d",&n) != EOF){
		ini(n);
		for (i = 1; i <= n; ++i)
			scanf("%d",&happy[i]);
		rt = -1;
		while (scanf("%d %d",&l,&k) && l+k){
			child[k].push_back(l);
			if (rt == -1 || l == rt)
				rt = k;
		}
		dfs(rt);
		printf("%d\n",max(dp[rt][0],dp[rt][1]));
	}
	return 0;
}