树形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