码迷,mamicode.com
首页 > 其他好文 > 详细

没有上司的舞会 树形DP

时间:2019-05-08 21:30:22      阅读:125      评论:0      收藏:0      [点我收藏+]

标签:节点   选择   c++   span   font   using   需要   div   color   

 

 

题意:有n个职员可以参加舞会,每个职员有一个欢乐值,职员之间就像一颗树,每个父节点都是子节点的上司,同时一个职员不可以他的直接上司一起参加,。现在选一些员工参加舞会,求参加员工的可能最大快乐值。

做法:设f[i][0]表示i代表的子树之下,i职员不参加的最大快乐值,f[i][1]表示i代表的子树之下,i职员参加的最大快乐值。则状态转移方程得

f[i][0]+=max(f[j][1],f[i][0])如果i不参加,那么他的每个儿子j可以选择参加或者不参加,累加每个儿子的较大的值。

f[i][1]+=f[j][0]如果i参加,那么它需要累加它每个儿子不参加时的最大快乐值。

然后就是从顶头上司根节点开始,深度优先遍历即可

#include<bits/stdc++.h>
using namespace std;
int f[1005][2];
int val[1005];
vector<int> v[1005];
int ans;
int vis[1005];

void dp(int t)
{
    f[t][0]=0;
    f[t][1]=val[t];
    for(int i=0;i<v[t].size();i++)
    {
        int y=v[t][i];
        dp(y);
        f[t][0]+=max(f[y][1],f[y][0]);
        f[t][1]+=f[y][0];           
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&val[i]);
    for(int i=1;i<=n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        v[b].push_back(a);
        vis[a]=1;
    }
    int root;
    for(int i=1;i<=n;i++)
    {
        if(vis[i]==0)
        {
            root=i;break;
        }
    }
    dp(root);
    printf("%d",max(f[root][1],f[root][0]));
}

 

没有上司的舞会 树形DP

标签:节点   选择   c++   span   font   using   需要   div   color   

原文地址:https://www.cnblogs.com/dongdong25800/p/10834719.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!