BZOJ3252 攻略 题解

BZOJ3252 攻略

题意

给出一棵树，最大化根节点到 $k$ 个叶子节点的点权。

解析

对这棵树进行长链剖分，每次选择权之总和最大的长链，这样子做既保证的权值最大，有保证了长链互不相交，具体证明方法见
CF526G 的前半段

附代码：

#include<bits/stdc++.h>
#define INF 1e16
#define endl '\n'
#define PI acos(-1)
#define pb push_back
#define int long long
#define mem(x,v) memset(x,v,sizeof x)
using namespace std;
const int N = 2e5+5;
inline int read(){
	register int x=0,f=1;register char c=getchar();
	while(c>'9'||c<'0'){if(c=='-')f=-1;c=getchar();}
	while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
	return x*f;
}

priority_queue <int> q;
int n,m,rt,cnt,ans;
int a[N],in[N],head[N],fa[N],mx[N],son[N];
struct Edge{int to,next;}edge[N<<1];
void add(int u,int v){edge[++cnt]={v,head[u]};head[u]=cnt;}

void dfs1(int u,int f){
	fa[u]=f;
	for(int i=head[u];i;i=edge[i].next){
		int v=edge[i].to;
		dfs1(v,u);
		if(mx[v]>mx[son[u]])son[u]=v;
	}
	mx[u]=a[u]+mx[son[u]];
}
void dfs2(int u,int top){
	if(u!=top)a[u]+=a[fa[u]];
	if(!son[u]){q.push(a[u]);return;}
	dfs2(son[u],top);
	for(int i=head[u];i;i=edge[i].next){
		int v=edge[i].to;
		if(v==son[u])continue;
		dfs2(v,v);
	}
}
signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1,u,v;i<n;i++)cin>>u>>v,add(u,v),in[v]=1;
	for(int i=1;i<=n;i++)if(!in[i])rt=i;
	dfs1(rt,0),dfs2(rt,rt);
	for(int i=1;i<=m&&(!q.empty());i++)ans+=q.top(),q.pop();
	cout<<ans;
	return 0;
}