树形DP的关键在于如何处理递归返回的信息。这题dp[i][0]表示不选i点时当前最高权值。dp[i][1]表示选i点时当前最高权值。状态转移方程:dp[u][[0]+=max(dp[v][0],dp[v][1]),dp[u][1]+=dp[v][0];
代码如下:
#include <iostream> #include <string.h> #include <math.h> #include <queue> #include <algorithm> #include <stdlib.h> #include <map> #include <set> #include <stdio.h> using namespace std; #define LL __int64 #define pi acos(-1.0) const int mod=100000000; const int INF=0x3f3f3f3f; const double eqs=1e-8; int dp[7000][2], c[7000], vis[7000]; int head[7000], cnt, n; struct node { int u, v, next; }edge[100000]; void add(int u, int v) { edge[cnt].v=v; edge[cnt].next=head[u]; head[u]=cnt++; } void dfs(int u) { vis[u]=1; dp[u][1]=c[u]; for(int i=head[u];i+1;i=edge[i].next){ int v=edge[i].v; if(!vis[v]){ dfs(v); dp[u][0]+=max(dp[v][0],dp[v][1]); dp[u][1]+=dp[v][0]; } } } void init() { memset(vis,0,sizeof(vis)); memset(head,-1,sizeof(head)); cnt=0; memset(dp,0,sizeof(dp)); } int main() { int i, j, u, v; while(scanf("%d",&n)!=EOF){ for(i=1;i<=n;i++){ scanf("%d",&c[i]); } init(); while(scanf("%d%d",&u,&v)!=EOF&&u&&v){ add(u,v); add(v,u); } dfs(1); printf("%d\n",max(dp[1][0],dp[1][1])); } return 0; }
HDU 1520 Anniversary party (树形DP)
原文地址:http://blog.csdn.net/scf0920/article/details/42783385