【DBSDFZOJ 4846】攻略（贪心）

我们可以得到一个简单的贪心策略。假设攻略是有顺序的，每次攻略必然选择当前价值和最大的路线。
我们可以维护每个节点到一个叶子节点的点权和的最大值，用大根堆维护，每次删去堆顶元素所代表的那条路线，并把这个路线旁的其它节点加入堆中。

#include<bits/stdc++.h>
#define N 200005
#define pa pair<long long,int>
using namespace std;
int n,k,head[N],nex[N],to[N],tp;
long long a[N],dis[N],ans;
priority_queue<pa>q;
inline void add(int x,int y){
    nex[++tp]=head[x];
    head[x]=tp;
    to[tp]=y;
}
void dfs(int x){
    dis[x]=a[x];
    for(int i=head[x];i;i=nex[i]){
        dfs(to[i]);
        dis[x]=max(dis[x],dis[to[i]]+a[x]);
    }
}
void solve(int x){
    bool flag=false;
    for(int i=head[x];i;i=nex[i]){
        if(dis[x]==dis[to[i]]+a[x] && !flag)    flag=true,solve(to[i]);
        else q.push(pa(dis[to[i]],to[i]));
    }
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;++i)    scanf("%lld",&a[i]);
    for(int i=1;i<n;++i){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
    }
    dfs(1); q.push(pa(dis[1],1));
    while(k--){
        ans+=q.top().first;
        solve(q.top().second);
        q.pop();
    }
    printf("%lld\n",ans);
    return 0;
}