題目連結

貪心，每次選價值最大的一條到根的鏈。比較顯然（不選白不選）。
考慮如何維護這個過程。一個點的價值選了就沒有了，而它只會影響它子樹裡的點，可以用DFS序+線段樹修改。而求最大值也可以用線段樹。
每個點只會被取一次，即價值也只會被清空一次。所以每選一條鏈就暴力往上跳，直到到一個清空過的點，順便線上段樹上修改經過點的子樹權值就可以了。
複雜度\(O((n+k)\log n)\)。

實際上，每個點只被統計一次，就是選\(k\)條最長的不相交的鏈（鍊形態是從上到下的）。
所以可以想到長鏈剖分。以路徑權值和作為深度，選最長的\(k\)條長鏈就行了。
用nth_element就可以把排序的複雜度也省了233.
所以複雜度\(O(n)\)

//10236kb   364ms
#include <cstdio>
#include <cctype>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 300000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=2e5+5;

int Enum,H[N],nxt[N],to[N],A[N],son[N],cnt;
LL mxd[N],val[N];
char IN[MAXIN],*SS=IN,*TT=IN;

inline int read()
{
    int now=0;register char c=gc();
    for(;!isdigit(c);c=gc());
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now;
}
inline void AE(int v,int u)
{
    to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
}
void DFS1(int x)
{
    LL mx=0;
    for(int i=H[x],v; i; i=nxt[i])
        DFS1(v=to[i]), mxd[v]>mx&&(mx=mxd[v],son[x]=v);
    mxd[x]=mx+A[x];
}
void DFS2(int x,int top)
{
    if(x==top) val[++cnt]=mxd[x];
    if(!son[x]) return;
    DFS2(son[x],top);
    for(int i=H[x]; i; i=nxt[i])
        if(to[i]!=son[x]) DFS2(to[i],to[i]);
}

int main()
{
    int n=read(),K=read();
    for(int i=1; i<=n; ++i) A[i]=read();
    for(int i=1; i<n; ++i) AE(read(),read());
    DFS1(1), DFS2(1,1);
    K=std::min(K,cnt), std::nth_element(val+1,val+cnt-K+1,val+1+cnt);
    LL ans=0;
    for(int i=cnt-K+1; i<=cnt; ++i) ans+=val[i];
    printf("%lld\n",ans);

    return 0;
}