【题解】bzoj3252 攻略

bzoj3252 攻略

\(\text{Solution:}\)

题目是指每次选择一条链就会把它清空，求选 \(k\) 条链的最大价值。

打眼看上去不知道咋做……想一下这不就是个长链剖分！

维护一个带权的长链剖分，然后直接对每一条链的权值和排序，取前 \(k\) 大就可以了。

因为长链剖分的每一条链都是不重复的，没有公共部分，恰好对应了删掉之后的链；由于我们按照权值做的带权长剖，所以它也满足贪心的性质。

这样就直接用长链剖分把贪心优化掉了。

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int MAXN=2e5+10;
int head[MAXN],tot,n,top[MAXN];
int len[MAXN],son[MAXN],pa[MAXN];
int val[MAXN],k;
vector<int>vec;
struct E{int nxt,to;}e[MAXN<<1];
inline void add(int x,int y){e[++tot]=(E){head[x],y};head[x]=tot;}
void dfs1(int x,int fa){
	pa[x]=fa;
	for(int i=head[x];i;i=e[i].nxt){
		int j=e[i].to;
		if(j==fa)continue;
		dfs1(j,x);
		if(len[j]>len[son[x]])son[x]=j;
	}
	len[x]=len[son[x]]+val[x];
}
void dfs2(int u,int t){
	top[u]=t;
	if(son[u])dfs2(son[u],t);
	for(int i=head[u];i;i=e[i].nxt){
		int j=e[i].to;
		if(j==pa[u]||j==son[u])continue;
		dfs2(j,j);
	}
}
signed main(){
	scanf("%lld%lld",&n,&k);
	for(int i=1;i<=n;++i)scanf("%lld",&val[i]);
	for(int i=1;i<n;++i){
		int u,v;
		scanf("%lld%lld",&u,&v);
		add(u,v);add(v,u);
	}
	dfs1(1,0);
	dfs2(1,1);
	for(int i=1;i<=n;++i)if(top[i]==i)vec.push_back(-len[i]);
//	for(int i=1;i<=n;++i)printf("%d ",len[i]);
//	cout<<endl;
	sort(vec.begin(),vec.end());
	int ans=0;
	for(int i=0;i<k;++i)ans-=vec[i];
	cout<<ans<<endl;
	return 0;
}