bzoj 3252攻略

Posted 2021-02-15 asuldb

题意

我们想到一个贪心，就是每次找到根路径前缀和最大的一个点，取走这条路径，同时把这条路径上的点权变成(0)

正确性显然

进一步发现我们需要从树上选择(m)条链使得链的总和最大

于是我们考虑换上长链剖分，长儿子定义为往下走点权最大的儿子，每次把最长的路径取走就好了

来一个堆维护一下就好了

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define re register
#define LL long long
#define mp std::make_pair
typedef std::pair<LL,int> pii;
const int maxn=200006;
inline int read() {
    char c=getchar();int x=0;while(c<'0'||x>'9') c=getchar();
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
std::priority_queue<pii> q;
struct E{int v,nxt;}e[maxn];
int head[maxn],a[maxn],son[maxn],vis[maxn];LL len[maxn];
int n,m,num;
inline void add(int x,int y) {
    e[++num].v=y;e[num].nxt=head[x];head[x]=num;
}
void dfs1(int x) {
    for(re int i=head[x];i;i=e[i].nxt) {
        dfs1(e[i].v);
        if(len[e[i].v]>len[son[x]]) son[x]=e[i].v;
    }
    len[x]=len[son[x]]+a[x];
}
int main() {
    n=read(),m=read();
    for(re int i=1;i<=n;i++) a[i]=read();
    for(re int x,y,i=1;i<n;i++) {
        x=read(),y=read(),add(x,y);
    }
    dfs1(1);
    for(re int i=1;i<=n;i++) q.push(mp(len[i],i));
    int tot=0;LL ans=0;
    while(tot<m) {
        int k=q.top().second;q.pop();
        if(vis[k]) continue;
        ans+=len[k];tot++;
        while(k&&!vis[k]) vis[k]=1,k=son[k];
    }
    printf("%lld
",ans);
    return 0;
}