HDU 1520

又是一眼题撒,树形DP的干活。

设dp[i][0]表示不邀请i的答案,dp[i][1]表示邀请i的答案。

那么dp[i][0]+=max(dp[son[i]][0],dp[son[i][1]]);下属可来可不来。

dp[i][1]+=(dp[son[i]][1]);下属必须不能来。

%%%__debug大神。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
const int MAXN=6001;
int plea[MAXN],dp[MAXN][2],n,root,fa[MAXN];
vector<int>G[MAXN];
void init()
{
	for(int i=0;i<MAXN;i++)if(G[i].size())G[i].clear();
	memset(fa,-1,sizeof(fa));
	memset(dp,0,sizeof(dp));
}
void dfs(int u,int fa)
{
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(v==fa)continue;
		dfs(v,u);
	}
	for(int i=0;i<G[u].size();i++)
	{
		int v=G[u][i];
		if(v==fa)continue;
		dp[u][0]+=max(dp[v][0],dp[v][1]);
		dp[u][1]+=dp[v][0];
	}
}
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		init();		
		for(int i=1;i<=n;i++)scanf("%d",&plea[i]),dp[i][1]+=plea[i];
		for(int i=1;i<=n;i++)
		{
			int a,b;
			scanf("%d %d",&a,&b);
			if(a==0&&b==0)break;
			G[b].push_back(a);
			fa[a]=b;
		}
		for(int i=1;i<=n;i++)if(fa[i]==-1){root=i;break;}
		dfs(root,-1);
		printf("%d\n",max(dp[root][0],dp[root][1]));
	}
}

你可能感兴趣的:(HDU)