樹形DP入門（POJ2342）

#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
const int N=6666;
int fa[N],dp[N][2],n;
void treedp(int rt)
{
	for(int i=1;i<=n;i++)
	{
		if(fa[i]==rt)
		{
			treedp(i);
			dp[rt][1]+=dp[i][0];
			dp[rt][0]+=max(dp[i][0],dp[i][1]);
		}
	}
}
int main()
{
	int i,a,b,ans,rt;
	while(scanf("%d",&n)!=EOF)
	{
		memset(dp,0,sizeof(dp));
		memset(fa,0,sizeof(fa));
		ans=0;
		for(i=1;i<=n;i++)
			scanf("%d",&dp[i][1]);
		while(scanf("%d%d",&a,&b)&&(a||b))
		{
			fa[a]=b;
			rt=b;
		}
		while(fa[rt])
			rt=fa[rt];
		treedp(rt);
		printf("%d\n",max(dp[rt][0],dp[rt][1]));
	}
	return 0;
}