HDU 1520 Anniversary party-----树形DP

dp[i][0]表示  i点 不取

dp[i][1]表示  i点取

j为i的子节点

则  dp[i][0] += max(dp[j][0],dp[j][1]);

      dp[i][1] += dp[j][0];

(j 子节点有很多)

如果 a 为根节点，结果就是 max(dp[a][0],dp[a][1]);  

用 vector 写得邻接表。。

#include<stdio.h>#include<string.h>#include<stdlib.h>#include<algorithm>#include<vector>using namespace std;vector<int> adj[6050];int h[6001],dp[6001][2],f[6050];void dfs(int p){dp[p][1]=h[p];int len=adj[p].size();for(int i=0;i<len;i++){dfs(adj[p][i]);}for(int i=0;i<len;i++){dp[p][0]+=max(dp[adj[p][i]][0],dp[adj[p][i]][1]);//重要！！dp[p][1]+=dp[adj[p][i]][0];}}int main(){int n;while(scanf("%d",&n)!=EOF){int i,j,k;for(i=1;i<=n;i++){scanf("%d",&h[i]);f[i]=-1;adj[i].clear();dp[i][0]=0;dp[i][1]=0;}int a,b;while(scanf("%d%d",&a,&b),a||b){f[a]=b;adj[b].push_back(a);}a=1;while(f[a]!=-1) a=f[a];dfs(a);printf("%d\n",max(dp[a][0],dp[a][1]));}}