HDU 1520

来源：互联网发布：网易邮箱客户端mac版编辑：程序博客网时间：2024/06/04 17:49

又是一眼题撒，树形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]));}}

0 0