poj2342 树形dp入门

来源:互联网 发布:javascript super 编辑:程序博客网 时间:2024/04/29 17:42

题意:

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

分析:

毫无疑问肯定需要先建树,然后就是从这棵树里面选择一些点,当然不能暴力。我们用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 100000005int 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]表示选择ivoid 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 ;}


0 0
原创粉丝点击