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
4 3 2 1 1
1 2
1 5
2 3
2 4
Sample Output
10
HINT
对于100%的数据，n<=200000，1<=场景价值<=2^31-1
很容易得到这样一个贪心策略：每次优先找一条从根开始的权值最大的路径，然后用数据结构修改其他路径的权值。
因为一个场景最多被选1次，所以每次顺着路径的末节点向上一直修改到上一个被选择的节点，这样修改的时间复杂度就是O(N)*O(数据结构)。
因为场景会影响一个子树内所有的节点，所以这里的数据结构用棵线段树就行了，时间复杂度为O(NlogN)。
#include<cstdio>
#include<cctype>
#include<queue>
#include<cstring>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
const int BufferSize=1<<16;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
if(head==tail) {
int l=fread(buffer,1,BufferSize,stdin);
tail=(head=buffer)+l;
}
return *head++;
}
inline int read() {
int x=0,f=1;char c=Getchar();
for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1;
for(;isdigit(c);c=Getchar()) x=x*10+c-'0';
return x*f;
}
typedef long long ll;
const int maxn=200010;
int n,k,A[maxn],first[maxn],next[maxn],to[maxn],e;
void AddEdge(int v,int u) {to[++e]=v;next[e]=first[u];first[u]=e;}
int st[maxn],en[maxn],pos[maxn],fa[maxn],del[maxn],cnt;
ll val[maxn];
void dfs(int x) {
st[x]=++cnt;pos[cnt]=x;val[x]+=A[x];
ren val[to[i]]=val[x],fa[to[i]]=x,dfs(to[i]);
en[x]=cnt;
}
struct Ans {int p;ll v;}maxv[maxn<<2];
Ans better(Ans A,Ans B) {
if(A.v>B.v) return A;
return B;
}
ll addv[maxn<<2];
void maintain(int o,int l,int r) {
int lc=o<<1,rc=lc|1;
if(l<r) maxv[o]=better(maxv[lc],maxv[rc]);
else maxv[o]=(Ans){l,0};
if(addv[o]) maxv[o].v+=addv[o];
}
void build(int o,int l,int r) {
if(l==r) addv[o]=val[pos[l]];
else {
int mid=l+r>>1,lc=o<<1,rc=lc|1;
build(lc,l,mid);build(rc,mid+1,r);
}
maintain(o,l,r);
}
void update(int o,int l,int r,int ql,int qr,int v) {
if(ql<=l&&r<=qr) addv[o]+=v;
else {
int mid=l+r>>1,lc=o<<1,rc=lc|1;
if(addv[o]) addv[lc]+=addv[o],addv[rc]+=addv[o],addv[o]=0;
if(ql<=mid) update(lc,l,mid,ql,qr,v);
else maintain(lc,l,mid);
if(qr>mid) update(rc,mid+1,r,ql,qr,v);
else maintain(rc,mid+1,r);
}
maintain(o,l,r);
}
int main() {
n=read();k=read();
rep(i,1,n) A[i]=read();
rep(i,1,n-1) AddEdge(read(),read());
dfs(1);build(1,1,n);ll ans=0;
while(k--) {
Ans x=maxv[1];ans+=x.v;
if(!x.v) break;
int u=pos[x.p];
while(u&&!del[u]) {
del[u]=1;
update(1,1,n,st[u],en[u],-A[u]);
u=fa[u];
}
}
printf("%lld\n",ans);
return 0;
}