POJ 2342 树形DP入门题

有一个大学的庆典晚会，想邀请一些在大学任职的人来参加，每个人有自己的搞笑值，但是现在遇到一个问题就是如果两个人之间有直接的上下级关系，那么他们中只能有一个来参加，求请来一部分人之后，搞笑值的最大是多少。

树形DP入门题。

DP部分：

dp[i][0]表示职员i不来参加party，以i为根的子树的最大搞笑值，

dp[i][1]表示职员i来参加party，以i为根的子树的最大搞笑值。

转移方程：

dp[cur][1]+=dp[next][0];
dp[cur][0]+=Max(dp[next][1],dp[next][0]);

#include "stdio.h"#include "string.h"#include "vector"using namespace std;struct node{    int fa;    vector<int>child;}data[6010];int dp[6010][2],vis[6010];int Max(int a,int b){    if (a<b) return b;else return a;}void dfs(int cur){    int i,next;    vis[cur]=1;    for (i=0;i<data[cur].child.size();i++)    {        next=data[cur].child[i];        if (vis[next]==0)        dfs(next);        dp[cur][1]+=dp[next][0];        dp[cur][0]+=Max(dp[next][1],dp[next][0]);    }}int main(){    int n,i,a,b;    while (scanf("%d",&n)!=EOF)    {        memset(dp,0,sizeof(dp));        memset(data,0,sizeof(data));        memset(vis,0,sizeof(vis));        for (i=1;i<=n;i++)            scanf("%d",&dp[i][1]);        while(scanf("%d%d",&a,&b))        {            if (a+b==0) break;            data[a].fa=b;            data[b].child.push_back(a);        }        for (i=1;i<=n;i++)            if (data[i].fa==0)            {                dfs(i);                break;            }        printf("%d\n",Max(dp[i][1],dp[i][0]));    }    return 0;}