【HAOI2015】【树形dp】树上染色

这是一道关于树形动态规划的问题,目标是在一棵点数为N的树上选择K个点染成黑色,其余点染成白色,以最大化黑点与黑点、白点与白点之间的距离之和。输入包括树的结构和K的值,输出是最大收益。解题策略是考虑每个子树及其与父节点边的贡献,将问题转化为树形背包问题进行求解。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

【题目描述】
有一棵点数为N的树,树边有边权。给你一个在0~N之内的正整数K,你要在这棵树中选择K个点,将其染成黑色,并将其他的N-K个点染成白色。将所有点染色后,你会获得黑点两两之间的距离加上白点两两之间距离的和的收益。问收益最大值是多少。
【输入格式】
第一行两个整数N,K。
接下来N-1行每行三个正整数fr,to,dis,表示该树中存在一条长度为dis的边(fr,to)。输入保证所有点之间是联通的。
【输出格式】
输出一个正整数,表示收益的最大值。
【输入样例1】
3 1
1 2 1
1 3 2
【输出样例1】
3
【输入样例2】
5 2
1 2 3
1 5 1
2 3 1
2 4 2
【输出样例2】
17
【样例解释】
在第二个样例中,将点1,2染黑就能获得最大收益。
【数据范围】
对于30%的数据,N<=20
对于50%的数据,N<=100
对于100%的数据,N<=2000,0<=K<=N。
题解:
其实这个题只需要考虑每个子树和他父亲的那条边对答案做了多少贡献即可(也就是子树内的白点数乘上子树外的白点数再加上子树内的黑点数乘上子树外的黑点数,最后再乘上边权),这个问题是可以分割成子问题的,对于每个子问题我们就可以枚举父节点给他的每个儿子分多少个黑点,然后就成了树形背包了。

#include<iostream>
#include<cstdio>
#include<vector>
using 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值