Dfsbzoj3252攻略

Posted -guz

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Dfsbzoj3252攻略相关的知识,希望对你有一定的参考价值。

Description

题目简述:树版[k取方格数]

众所周知,桂木桂马是攻略之神,开启攻略之神模式后,他可以同时攻略k部游戏。

今天他得到了一款新游戏《XX半岛》,这款游戏有n个场景(scene),某些场景可以通过不同的选择支到达其他场景。所有场景和选择支构成树状结构:开始游戏时在根节点(共通线),叶子节点为结局。每个场景有一个价值,现在桂马开启攻略之神模式,同时攻略k次该游戏,问他观赏到的场景的价值和最大是多少(同一场景观看多次是不能重复得到价值的)

“为什么你还没玩就知道每个场景的价值呢?”

“我已经看到结局了。”

Input

第一行两个正整数n,k

第二行n个正整数,表示每个场景的价值

以下n-1行,每行2个整数a,b,表示a场景有个选择支通向b场景(即a是b的父亲)

保证场景1为根节点

n<=200000,1<=场景价值<=2^31-1

Output

输出一个整数表示答案

如果我们的图是这样

技术分享图片

显然,这样我们会选择(8),而不会选择(6,5)

此时我们按照价值划分为长短链,当前的父亲节点(u)就带有价值(8),这个时候直接塞过去就好.

此时就可以等价为我们拆成了这样

技术分享图片

此时,我们的答案就转化为求这些链的权值的前(k)大的和.

代码

#include<cstdio>
#include<cctype>
#include<algorithm>
#define int long long
#define R register
using namespace std;
inline void in(int &x)
{
    int f=1;x=0;char s=getchar();
    while(!isdigit(s)){if(s=='-')f=-1;s=getchar();}
    while(isdigit(s)){x=x*10+s-'0';s=getchar();}
    x*=f;
}
int n,k,val[200008],stk[200008],cnt,top,ans;
int head[200008],tot,son[200008];
struct cod{int u,v;}edge[500008];
inline void add(int x,int y)
{
    edge[++tot].u=head[x];
    edge[tot].v=y;
    head[x]=tot;
}
void dfs(int u,int fa)
{
    for(R int i=head[u];i;i=edge[i].u)
    {
        if(edge[i].v==fa)continue;
        dfs(edge[i].v,u);
        if(val[edge[i].v]>val[son[u]])
            son[u]=edge[i].v;
    }
    val[u]+=val[son[u]];
    for(R int i=head[u];i;i=edge[i].u)
    {
        if(edge[i].v==fa or edge[i].v==son[u])continue;
        if(val[edge[i].v]>val[u])
            swap(val[edge[i].v],val[u]);
        stk[++top]=val[edge[i].v];
    }
}
signed main()
{
    in(n),in(k);
    for(R int i=1;i<=n;i++)in(val[i]);
    for(R int i=1,x,y;i<n;i++)
    {
        in(x),in(y);
        add(x,y);add(y,x);
    }
    dfs(1,0);
    stk[++top]=val[1];
    sort(stk+1,stk+top+1);
    for(R int i=top;i>=0;i--)
    {
        cnt++;
        ans+=stk[i];
        if(cnt==k)break;
    }
    printf("%lld",ans);
}

以上是关于Dfsbzoj3252攻略的主要内容,如果未能解决你的问题,请参考以下文章

BZOJ3252: 攻略

线段树 BZOJ3252 攻略

BZOJ-3252攻略 DFS序 + 线段树 + 贪心

[HYSBZ - 3252] 攻略

BZOJ_3252_攻略_线段树+dfs序

bzoj3252