POJ 2342 树形DP入门水题

题意

N个人，每个人有一个参加宴会的开心值，每个人不能和他的上级同时参加宴会。问参加宴会的人员开心值总和最大是多少？

题解

最基础的树形DP，对于一个人有两种状态，dp[m][0]代表不参加宴会，则dp[m][0]+=max(dp[m][1],dp[m][0])。dp[m][1]代表参加宴会，因此其子节点就不能参加宴会，dp[m][1]+=dp[m][0]。

代码

#include <iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

using namespace std;
int dp[6010][2];
int p[6010];
bool vis[6010];
int n;

void dfs(int m){
vis[m]=true;
for(int i=1;i<=n;i++){
if(!vis[i]&&p[i]==m){
dfs(i);
dp[m][0]+=max(dp[i][1],dp[i][0]);
dp[m][1]+=dp[i][0];
}
}
}

int main()
{
while(~scanf("%d",&n)){
memset(dp,0,sizeof(dp));
memset(p,0,sizeof(p));
memset(vis,false,sizeof(vis));
for(int i=1;i<=n;i++){
scanf("%d",&dp[i][1]);
}
int n,m;
while(scanf("%d%d",&n,&m)){
if(n+m==0)
break;
p
=m;
}
dfs(0);
printf("%d\n",max(dp[0][1],dp[0][0]));
}
return 0;
}