P1352 没有上司的舞会

P1352 没有上司的舞会

题意:

给你一个树,每个点都有权值,选择一些点使得权值和最大,要求父亲节点和子节点不能同时选择

题解:

经典树形dp
dp[x][0]表示以x为根的子树,且x不参加舞会的最大快乐值
dp[x][1]表示以x为根的子树,且x参加了舞会的最大快乐值
则dp[x][0] = ∑{ max(dp[y][0],dp[y][1]) } (y是x的儿子)
dp[x][1] = ∑{ dp[y][0] } + a[x] (y是x的儿子)
找到唯一的树根root
ans=max(dp[root][0],dp[root][1])

代码:

#include<bits/stdc++.h>
using namespace std;
#define MAXN 6005
int a[MAXN];
int v[MAXN];
vector<int> son[MAXN];
int f[MAXN][2];
void dp(int x)
{
    f[x][0]=0;
    f[x][1]=a[x];
    for(int i=0;i<son[x].size();i++)
    {
        int y=son[x][i];
        dp(y);
        f[x][0]+=max(f[y][0],f[y][1]);
        f[x][1]+=f[y][0];
    }
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n-1;i++)
    {
        int x,y;
        cin>>x>>y;
        son[y].push_back(x);
        v[x]=1;
    }
    int root;
    for(int i=1;i<=n;i++)
    if(!v[i]) {root=i;break;}
    dp(root);
    cout<<max(f[root][0],f[root][1])<<endl;
    return 0;
}

全部评论

相关推荐

05-12 11:09
已编辑
门头沟学院 后端
已注销:没必要放这么多专业技能的描述。这些应该是默认已会的,写这么多行感觉在凑内容。项目这块感觉再包装包装吧,换个名字,虽然大家的项目基本都是网上套壳的,但是你这也太明显了。放一个业务项目,再放一个技术项目。技术项目,例如中间件的一些扩展和尝试。
点赞 评论 收藏
分享
后来123321:别着急,我学院本大二,投了1100份,两个面试,其中一个还是我去线下招聘会投的简历,有时候这东西也得看运气
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务