DP - 树形DP - 二叉苹果树
有一棵二叉苹果树,如果树枝有分叉,一定是分两叉,即没有只有一个儿子的节点。
这棵树共 N 个节点,编号为 1 至 N,树根编号一定为 1。
我们用一根树枝两端连接的节点编号描述一根树枝的位置。
一棵苹果树的树枝太多了,需要剪枝。但是一些树枝上长有苹果,给定需要保留的树枝数量,求最多能留住多少苹果。
这里的保留是指最终与1号点连通。
输入格式
第一行包含两个整数 N 和 Q,分别表示树的节点数以及要保留的树枝数量。
接下来 N−1 行描述树枝信息,每行三个整数,前两个是它连接的节点的编号,第三个数是这根树枝上苹果数量。
输出格式
输出仅一行,表示最多能留住的苹果的数量。
数据范围
1≤Q<N≤100.
N≠1,
每根树枝上苹果不超过 30000 个。
输入样例:
5 2
1 3 1
1 4 10
2 3 20
3 5 20
输出样例:
21
分析:
容 易 看 出 类 似 于 一 个 树 上 分 组 背 包 问 题 。 本 题 需 要 注 意 的 是 , 1 号 节 点 必 须 保 留 , 也 就 是 必 须 保 留 一 条 与 1 号 节 点 直 接 相 连 的 边 。 容易看出类似于一个树上分组背包问题。\\本题需要注意的是,1号节点必须保留,也就是必须保留一条与1号节点直接相连的边。 容易看出类似于一个树上分组背包问题。本题需要注意的是,1号节点必须保留,也就是必须保留一条与1号节点直接相连的边。
我 们 可 以 把 每 条 边 的 权 值 视 作 这 条 边 上 孩 子 节 点 的 权 值 , 每 个 节 点 的 体 积 视 为 1 , 我们可以把每条边的权值视作这条边上孩子节点的权值,每个节点的体积视为1, 我们可以把每条边的权值视作这条边上孩子节点的权值,每个节点的体积视为1,
于 是 本 题 就 是 在 以 1 号 节 点 为 根 的 树 的 2 棵 子 树 中 选 择 Q − 1 个 节 点 , 使 得 收 益 最 大 。 于是本题就是在以1号节点为根的树的2棵子树中选择Q-1个节点,使得收益最大。 于是本题就是在以1号节点为根的树的2棵子树中选择Q−1个节点,使得收益最大。
但 是 选 择 每 个 节 点 都 要 保 证 该 点 与 1 号 点 是 连 通 的 。 但是选择每个节点都要保证该点与1号点是连通的。 但是选择每个节点都要保证该点与1号点是连通的。
那 么 从 1 号 点 向 下 选 择 时 , 选 择 每 个 节 点 都 要 先 选 择 其 父 节 点 。 那么从1号点向下选择时,选择每个节点都要先选择其父节点。 那么从1号点向下选择时,选择每个节点都要先选择其父节点。
状 态 表 示 f [ i , j ] : 表 示 以 i 为 根 的 树 , 保 留 j 条 边 的 最 大 收 益 。 状态表示f[i,j]:表示以i为根的树,保留j条边的最大收益。 状态表示f[i,j]:表示以i为根的树,保留j条边的最大收益。
状
态
计
算
:
状态计算:
状态计算:
说明:
对
节
点
i
而
言
,
将
其
两
个
孩
子
节
点
视
作
两
个
物
品
组
,
在
两
个
物
品
组
中
选
择
物
品
。
以
i
为
根
的
子
树
中
选
择
j
个
物
品
,
若
在
物
品
组
s
1
中
选
,
就
要
先
选
择
s
1
,
那
么
就
在
s
1
的
孩
子
中
选
择
j
−
1
个
。
对节点i而言,将其两个孩子节点视作两个物品组,在两个物品组中选择物品。\\以i为根的子树中选择j个物品,若在物品组s1中选,就要先选择s1,那么就在s1的孩子中选择j-1个。
对节点i而言,将其两个孩子节点视作两个物品组,在两个物品组中选择物品。以i为根的子树中选择j个物品,若在物品组s1中选,就要先选择s1,那么就在s1的孩子中选择j−1个。
代码:
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=110,M=2*N;
int n,m;
int e[M],ne[M],w[M],h[N],idx;
int f[N][N]; //f[i][j]:从以i为根的子树种,选j条边的最大价值
void add(int a,int b,int c)
{
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u,int fa)
{
for(int i=h[u];~i;i=ne[i])
{
int v=e[i];
if(v==fa) continue;
dfs(v,u);
for(int j=m;j;j--)
for(int k=0;k<j;k++)
f[u][j]=max(f[u][j],f[u][j-k-1]+f[v][k]+w[i]);
}
}
int main()
{
memset(h,-1,sizeof h);
cin>>n>>m;
for(int i=0;i<n-1;i++)
{
int a,b,c;
cin>>a>>b>>c;
add(a,b,c),add(b,a,c);
}
dfs(1,-1);
cout<<f[1][m]<<endl;
return 0;
}