B - Long Live the Queen SGU - 143

题意:给你n个节点,每个节点都有一个权值,还给你一些边,现在让你找一个非空子集,是的这个非空子集的和最大。

思路:使用树形dp,每个节点都有选和不选,在决定选还是不选的时候,把当前所有子树权值之和和0进行比较,如果是大于0,就选,如果是小于0就不选,这个题有一个坑点,题目要求是非空子集,有这样一种情况,所有的权值都是小于0的,这是也是必须要选一个的。

比赛的时候题意理解可能有点问题,把题意理解成不能n个节点全选了,郝老师说,前面那句话可能就是随口一提,只需要关注非空子集这一个条件即可。

#include <bits/stdc++.h>

using namespace std;

const int maxn=16010;
vector<int> v[maxn];
int dp[maxn];
const int ind=0x3f3f3f;
int a[maxn];
int n;
int  dfs(int x,int fa)
{
    dp[x]=a[x];
    for(int i=0; i<v[x].size(); i++)
    {
        int u=v[x][i];
        int temp=0;
        if(u!=fa) temp=dfs(u,x);
        dp[x]+=max(0,temp);
    }
    return dp[x];
}
int main()
{
    cin>>n;
    int ans=-10000;
    for(int i=1; i<=n; i++)
    {
        cin>>a[i];
    }
    for(int i=0; i<n-1; i++)
    {
        int x,y;
        cin>>x>>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    dfs(1,-1);
    for(int i=1; i<=n; i++)
    {
        ans=max(ans,dp[i]);
    }
    cout<<ans<<endl;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值