poj2342 树形dp入门

题意：

公司要举办一次晚会，但是为了使得晚会的气氛更加活跃，每个参加晚会的人都不希望在晚会中见到他的直接上司，现在已知每个人的活跃指数和上司关系，求邀请哪些人来能使得晚会的总活跃指数最大。

分析：

毫无疑问肯定需要先建树，然后就是从这棵树里面选择一些点，当然不能暴力。我们用dp[i][0]表示不选择第i棵树，他的子数所能取得的最大值，

dp[i][1]表示选取第i棵树的最大值。

然后我们就可以得到递推方程，dp[i][1] += dp[j][0]  //j是i的直接子节点     dp[i][0] += max(dp[j][0] , dp[j][1]) ;

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>

using namespace std ;
#define mem(a) memset(a,0,sizeof(a))
#define inf 100000005

int const maxn = 6005;
int n ;
bool vis[maxn];
int father[maxn];
//表示的是直接上下级关系，也就是在树中father[i]=j表示的是i是j的子树
int dp[maxn][2];
//dp[i][0]表示的是树建立好之后不选择i所能得到的i及i的子树的最大值
//dp[i][1]表示选择i

void dfs(int root)
{
    vis[root] = 1 ; //表示访问过了
    for(int i = 1 ; i <= n ; i++)
    {
        if(!vis[i]&&father[i]==root)  //找root的子节点
        {
            dfs(i);
            dp[root][1]+=dp[i][0];
            dp[root][0]+=max(dp[i][0],dp[i][1]);
        }
    }
}

int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        mem(dp);
        for(int i = 1 ; i <= n ; i++)
        {
            scanf("%d",&dp[i][1]);
        }
        mem(father);
        int root = 1 ; //表示的是这课树的根
        int l,k;
        while(scanf("%d%d",&l,&k)&&(l+k))
        {
            father[l]=k;
        }
        while(father[root])
        {
            root = father[root] ;  //查找父节点，没有父节点的就是根
        }
        mem(vis);
        dfs(root);
        printf("%d\n",max(dp[root][0],dp[root][1]));
    }
    return 0 ;
}