HDU - 1520 Anniversary party（树形DP）

HDU - 1520 Anniversary party（树形DP+线性DP）

#include<iostream>
#include<cstring>
#include<vector>
using namespace std;

const int N = 6010;
vector<int> g[N];
int val[N],dp[N][2],fa[N],n;

void dfs(int u)
{
	dp[u][0]=0,dp[u][1]=val[u];
	for(auto &v:g[u])
	{
		dfs(v);
		dp[u][0]+=max(dp[v][0],dp[v][1]);
		dp[u][1]+=dp[v][0]; 
	}
}

int main()
{
	while(cin>>n)
	{
		memset(fa,-1,sizeof fa);
		for(int i=1;i<=n;i++) cin>>val[i],g[i].clear();
		
		int a,b;
		while(cin>>a>>b,a+b)
		{
			g[b].push_back(a);
			fa[a]=b;
		}
		
		int root=1;
		while(fa[root]!=-1) root=fa[root];
		dfs(root); 
		cout<<max(dp[root][0],dp[root][1])<<endl;
	}
	return 0;
}