标签：bzoj3252 链剖分 int 题解 len son dfs2 MAXN 攻略
\(\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;
}
标签：bzoj3252,链剖分,int,题解,len,son,dfs2,MAXN,攻略 来源： https://www.cnblogs.com/h-lka/p/15058874.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。