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攻略的主要内容,如果未能解决你的问题,请参考以下文章