HDU1520 Anniversary party 树形DP

传送门：HDU1520

题意：学校要开一个聚会。学校的教职工之间有上下级关系，为了让所有人开心，宴会组织者决定不会同时邀请一个人和他的上级，对于每一个人，他能给聚会带来的欢乐度有一个值，问组织者该邀请哪些人能够使宴会的欢乐度达到最大值。

典型的树形DP，同时也是我第一个树形DP的题目。

看了题解以后发现树形DP不过是把DP换到了图上去做，所以要做的第一步就是建图，我采用的是做图论题目经常用的 链式前向星法建图，也可以用vector或者链表去建图。

然后是状态转移方程：我们可以用一个二维二态的数组来描述，dp[i][1]表示第i个参与者参加了，dp[i][0]表示第i个参与者没有参加。则dp[i][1]=dp[i][1]+dp[i-1][0]，dp[i][0]=dp[i][0]+Max(dp[i-1][0],dp[i-1][1])。

然后就是dfs跑就行了。

另外注意要从根节点开始dfs，其实这里建图也可以建无向图，然后就可以从任意点开始dfs了，不过还是要加个标记数组避免重复，并且建图的时候还稍稍麻烦些，详见邝斌巨巨博客。

代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<map>
#define ll long long
#define pi acos(-1)
#define inf 0x3f3f3f3f
using namespace std;
typedef pair<int,int>P;
int head[6010],cnt;
int w[6010],dp[6010][2];
bool book[6010];
struct node{
	int f,c;
	int next;
}tree[6010];
void init()
{
	cnt=0;
	memset(head,-1,sizeof(head));
	memset(book,0,sizeof(book));
 }
void addedge(int f,int c)
{
	tree[cnt].c=c;
	tree[cnt].f=f;
	tree[cnt].next=head[f];
	head[f]=cnt++;
}
void dfs(int k)
{
	dp[k][0]=0;
	dp[k][1]=w[k];
	for(int i=head[k];i!=-1;i=tree[i].next)
	{
		dfs(tree[i].c);//先确定下级的最优解
		dp[k][1]+=dp[tree[i].c][0];
		dp[k][0]+=max(dp[tree[i].c][0],dp[tree[i].c][1]);
	}
}
int main()
{
	int n;
	while(~scanf("%d",&n))
	{
		init();
		for(int i=1;i<=n;i++)
		scanf("%d",w+i);
		int a,b;
		while(scanf("%d%d",&a,&b)&&a+b)
		{
			addedge(b,a);
			book[a]=1;
		}
		int root;
		for(int i=1;i<=n;i++)//找树根
		if(!book[i])
		{
			root=i;
			break;
		}
		dfs(root);
		printf("%d\n",max(dp[root][0],dp[root][1]));
	}
    return 0;
}