HDU 1520 Anniversary party(树形DP入门)

d[i][0]表示i结点参加party而i结点儿子结点不参加 转移:d[i][0]=rating[i]+sum(dp[j][1]) i结点的rating加上所有的儿子结点不参加的值

d[i][1]表示i结点不参加而让儿子结点门选择参加或者不参加 转移d[i][1]=sum(max(dp[j][0],dp[j][1])) i结点不参加的值为所有儿子结点的参加或者不参加的最大值的和

#include <iostream>
#include <memory.h>
#include <cstdio>
using namespace std;
const int maxn=6100;
struct node
{
	int child,bros,rating,v1,v2;
}trees[maxn];
int n,ans,dg[maxn];
void dfs(int u){
	int chl=trees[u].child;
	trees[u].v1=trees[u].rating;
	while(chl){
		dfs(chl);
		trees[u].v1+=trees[chl].v2;//自己参加,儿子不参加
		trees[u].v2+=max(trees[chl].v1,trees[chl].v2);//自己不参加,儿子们选择参加不参加
		chl=trees[chl].bros;
	}
	ans=max(ans,max(trees[u].v1,trees[u].v2));//取两者的最大值
}
int main(){
	while (scanf("%d",&n)==1){
		memset(trees,0,sizeof(trees));
		memset(dg,0,sizeof(dg));
		for (int i=1;i<=n;++i){
			scanf("%d",&trees[i].rating);
		}
		int u,v;
		while (scanf("%d%d",&v,&u)){
			if(!u&&!v)break;
			dg[v]++;
			if(trees[u].child==0){
				trees[u].child=v;
			}else{
				int t=trees[u].child;
				while(trees[t].bros){
					t=trees[t].bros;
				}
				trees[t].bros=v;
			}
		}
		ans=0;
		for (int i=1;i<=n;++i){
			if(!dg[i]){//找根节点
				dfs(i);
			}
		}
		
		printf("%d\n",ans);
	}
	return 0;
}