【BZOJ3252】攻略（长链剖分，贪心）
题面
BZOJ
给定一棵树，每个点有点权，选定(k)个叶子，满足根到(k)个叶子的所有路径所覆盖的点权和最大。
题解
一个假装是对的贪心：
每次选择最大的路径，然后将路径上所有点的权值清零。
那么我们可以用长链剖分来实现这个贪心。
链长改为最大的路径权值和，这样子把每条重链的权值丢进一个堆里面取(k)次即可。
正确性自己(YY)一下，发现是对的
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
#define ll long long
#define MAX 200200
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
priority_queue<ll> Q;
struct Line{int v,next;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v){e[cnt]=(Line){v,h[u]};h[u]=cnt++;}
int a[MAX],n,K,hson[MAX];
ll ms[MAX],ans;
void dfs1(int u)
{
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;dfs1(v);
if(ms[v]>ms[hson[u]])hson[u]=v;
}
ms[u]+=a[u]+ms[hson[u]];
}
void dfs2(int u,int tp)
{
if(u==tp)Q.push(ms[u]);
if(hson[u])dfs2(hson[u],tp);
for(int i=h[u];i;i=e[i].next)
if(e[i].v!=hson[u])dfs2(e[i].v,e[i].v);
}
int main()
{
n=read();K=read();
for(int i=1;i<=n;++i)a[i]=read();
for(int i=1,x,y;i<n;++i)x=read(),y=read(),Add(x,y);
dfs1(1);dfs2(1,1);
while(K&&!Q.empty())ans+=Q.top(),Q.pop(),--K;
printf("%lld
",ans);
return 0;
}