3252: 攻略(长链刨分)

题意:给你一棵树,每个节点都有一个价值,但每个价值只能加一次,选择k次图上叶子节点到叶子节点,使得这k次经过的价值最大(注意不能重复)。
思路:找一个点当作根节点,长链刨分。
代码:

#include<bits/stdc++.h>
using namespace std;
const int N=200005;
typedef long long ll;
int head[N],nxt[N<<2],to[N<<2];
ll s[N];
int cnt=0;
void add(int u,int v)
{
    to[++cnt]=v;
    nxt[cnt]=head[u];
    head[u]=cnt;
}
int son[N];
ll value[N];
priority_queue<ll>Q;
//求价值
void dfs1(int x)
{
    for(int i=head[x];i;i=nxt[i])
    {
        if(to[i]!=x)
            dfs1(to[i]);
        if(value[to[i] ]>value[son[x] ])
            son[x]=to[i];
    }
    value[x]=value[son[x] ]+s[x];
}
//最大路径
void dfs2(int x,int top)
{
    if(x==top)Q.push(value[x]);
    if(son[x])dfs2(son[x],top);
    for(int i=head[x];i;i=nxt[i])
    {
        if(to[i]!=x && to[i]!=son[x])
            dfs2(to[i],to[i]);
    }
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
        scanf("%lld",&s[i]);
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
    }
    dfs1(1);
    dfs2(1,1);
    ll ans=0;
    while(k&&!Q.empty())ans+=Q.top(),Q.pop(),k--;
    printf("%lld",ans);
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值