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

 

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

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

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

“我已经看到结局了。”

第一行两个正整数n,k

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

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

保证场景1为根节点

输出一个整数表示答案

#include<iostream>

#include<cstdio>

#include<algorithm>

#include<cmath>

#include<cstring>

using namespace std;

int read()

{

    int x=,f=; char ch=getchar();

    while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}

    while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}

    return x*f;

}

#define MAXN 200010

#define LL long long

int N,K;

bool visit[MAXN];

LL sumV[MAXN],val[MAXN];

struct EdgeNode{int next,to;}edge[MAXN<<];

int head[MAXN],cnt=,fa[MAXN];

void AddEdge(int u,int v) {cnt++; edge[cnt].next=head[u]; head[u]=cnt; edge[cnt].to=v;}

void InsertEdge(int u,int v) {fa[v]=u; AddEdge(u,v); AddEdge(v,u);}

int pl[MAXN],pr[MAXN],dfn,pre[MAXN];

void DFS(int now,int last)

{

    pl[now]=++dfn; pre[dfn]=now;

    for (int i=head[now]; i; i=edge[i].next)

        if (edge[i].to!=last)

            sumV[edge[i].to]=sumV[now]+val[edge[i].to],DFS(edge[i].to,now);

    pr[now]=dfn;

}

struct SegmentTreeNode{int l,r,maxp; LL maxx,tag;}tree[MAXN<<];

int Maxp(SegmentTreeNode ls,SegmentTreeNode rs) {return ls.maxx>rs.maxx? ls.maxp:rs.maxp;}

void Update(int now)

{

    tree[now].maxx=max(tree[now<<].maxx,tree[now<<|].maxx);

    tree[now].maxp=Maxp(tree[now<<],tree[now<<|]);

}

void BuildTree(int now,int l,int r)

{

    tree[now].l=l; tree[now].r=r;

    if (l==r) {tree[now].maxx=sumV[pre[l]]; tree[now].maxp=l; return;}

    int mid=(l+r)>>;

    BuildTree(now<<,l,mid); BuildTree(now<<|,mid+,r);

    Update(now);

}

void PushDown(int now)

{

    if (tree[now].l==tree[now].r || !tree[now].tag) return;

    LL tag=tree[now].tag; tree[now].tag=;

    tree[now<<].maxx+=tag; tree[now<<|].maxx+=tag;

    tree[now<<].tag+=tag; tree[now<<|].tag+=tag;

}

void Change(int now,int L,int R,int val)

{

    PushDown(now);

    int l=tree[now].l,r=tree[now].r;

    if (L<=l && R>=r) {tree[now].maxx+=val; tree[now].tag+=val; return;}

    int mid=(l+r)>>;

    if (L<=mid) Change(now<<,L,R,val);

    if (R>mid) Change(now<<|,L,R,val);

    Update(now);

}

int main()

{

    N=read(); K=read();

    for (int i=; i<=N; i++) val[i]=read();

    for (int x,y,i=; i<=N-; i++) x=read(),y=read(),InsertEdge(x,y);

    sumV[]=val[]; DFS(,);

//    for (int i=1; i<=N; i++)

//        printf("ID=%d   [%d , %d] %d %d\n",pre[i],pl[pre[i]],pr[pre[i]],val[pre[i]],sumV[pre[i]]);

    LL ans=;

    BuildTree(,,N);

    for (int i=; i<=K; i++)

        {

//            printf("NOW=%d   :  %d , %d\n",i,tree[1].maxx,tree[1].maxp);

            if (tree[].maxx>) ans+=tree[].maxx; else break;

            for (int x=pre[tree[].maxp]; !visit[x]&&x; x=fa[x])

                visit[x]=,Change(,pl[x],pr[x],-val[x]);

        }

    printf("%lld\n",ans);

    return ;

}