题目描述
有一棵二叉苹果树,如果数字有分叉,一定是分两叉,即没有只有一个儿子的节点。这棵树共N个节点,标号1至N,树根编号一定为1。
我们用一根树枝两端连接的节点编号描述一根树枝的位置。一棵有四根树枝的苹果树,因为树枝太多了,需要剪枝。但是一些树枝上长有苹果,给定需要保留的树枝数量,求最多能留住多少苹果。
输入
第一行两个数N和Q,N表示树的节点数,Q表示要保留的树枝数量。
接下来N−1行描述树枝信息,每行三个整数,前两个是它连接的节点的编号,第三个数是这根树枝上苹果数量。
输出
输出仅一行,表示最多能留住的苹果的数量。
样例输入 Copy
5 2
1 3 1
1 4 10
2 3 20
3 5 20
样例输出 Copy
21
提示
对于100%的数据,1≤Q≤N≤100,N≠1,每根树枝上苹果不超过30000个。
2020.3.8更新
解析:
设f[i][j]表示以i为根,选j条边的最大值。
我们可以把m看成体积,每个点的体积为1。
把根看成分组,子树看成每个分组的物品----------------------->转化成分组背包
dfs搜一遍,跑一个分组背包
#include<bits/stdc++.h>
using namespace std;
const int N=330;
int f[N][N];
int h[N],e[N],ne[N],idx,w[N];
int n,m;
void add(int a,int b,int c)
{
e[idx]=b;w[idx]=c;ne[idx]=h[a];h[a]=idx++;
}
void dfs(int u,int father)
{
for(int i=h[u];i!=-1;i=ne[i])
{
if(e[i]==father) continue;
dfs(e[i],u);
for(int j=m;j>=0;j--)
for(int k=0;k<j;k++)
f[u][j]=max(f[u][j],f[u][j-k-1]+f[e[i]][k]+w[i]);
}
}
int main()
{
cin>>n>>m;
memset(h,-1,sizeof h);
for(int i=1;i<n;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;
}