标签：贪心，dfs
Description
题目简述：树版[k取方格数]
众所周知，桂木桂马是攻略之神，开启攻略之神模式后，他可以同时攻略k部游戏。
今天他得到了一款新游戏《XX半岛》，这款游戏有n个场景(scene)，某些场景可以通过不同的选择支到达其他场景。所有场景和选择支构成树状结构：开始游戏时在根节点（共通线），叶子节点为结局。每个场景有一个价值，现在桂马开启攻略之神模式，同时攻略k次该游戏，问他观赏到的场景的价值和最大是多少（同一场景观看多次是不能重复得到价值的）
“为什么你还没玩就知道每个场景的价值呢？”
“我已经看到结局了。”
Input
第一行两个正整数n,k
第二行n个正整数，表示每个场景的价值
以下n-1行,每行2个整数a,b，表示a场景有个选择支通向b场景（即a是b的父亲）
保证场景1为根节点
Output
输出一个整数表示答案
Sample Input
5 2
4 3 2 1 1
1 2
1 5
2 3
2 4
Sample Output
10
HINT
对于100%的数据，n<=200000，1<=场景价值<=2^31-1
Source
分析：这题是钟长者出的模拟题，但因为不明原因搞上了bzoj
很多人都是线段树+dfs序的做法
实际上并不需要
可以先dfs一遍，处理出w[x]表示以x为根节点的子树中取一条价值和最大的链
每次选择x中w[son]最大的一个转移：w[x]=max(w[son])+a[x]
然后将w排序，转化为c数组，取前k大
Code
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<cmath>
#define rep(i,a,b) for(register int i=a;i<=b;i++)
#define dep(i,a,b) for(register int i=a;i>=b;i--)
#define mem(x,num) memset(x,num,sizeof x)
#define v e[i].to
#define ll long long
#ifdef WIN32
#define LL "%I64d"
#else
#define LL "%lld"
#endif
using namespace std;
inline ll read()
{
ll f=1,x=0;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
const ll maxn=2e5+6;
ll head[maxn],a[maxn],n,k,w[maxn],cnt=0,c[maxn],tot=0;
struct edge{ll to,next;}e[maxn<<1];
#define v e[i].to
#define reg(x) for(int i=head[x];i;i=e[i].next)
inline bool cmp(ll x,ll y){return x>y;}
void dfs(int x,int fa)
{
ll Max=0,mx;
reg(x)
{
if(v==fa)continue;
dfs(v,x);
if(w[v]>Max)Max=w[v],mx=v;
}
w[x]=Max+a[x];
reg(x)
{
if(v==fa||v==mx)continue;
c[++tot]=w[v];
}
}
int main()
{
n=read(),k=read();
rep(i,1,n)a[i]=read();
rep(i,1,n-1){
int U=read(),V=read();
e[++cnt]=(edge){V,head[U]};head[U]=cnt;
e[++cnt]=(edge){U,head[V]};head[V]=cnt;
}
tot=0;dfs(1,1);
c[++tot]=w[1];
sort(c+1,c+1+tot,cmp);
ll ans=0;
k=min(k,tot);
rep(i,1,k)ans+=c[i];
cout<<ans<<endl;
return 0;
}
