传送门
题目描述
有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点)
这棵树共有N个结点(叶子点或者树枝分叉点),编号为1-N,树根编号一定是1。
我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有4个树枝的树
2 5
\ /
3 4
\ /
1
现在这颗树枝条太多了,需要剪枝。但是一些树枝上长有苹果。
给定需要保留的树枝数量,求出最多能留住多少苹果。
输入格式
第1行2个数,N和Q(1<=Q<= N,1<N<=100)。
N表示树的结点数,Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。
每行3个整数,前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。
每根树枝上的苹果不超过30000个。
输出格式
一个数,最多能留住的苹果的数量。
输入
5 2
1 3 1
1 4 10
2 3 20
3 5 20
输出
21
这个题是一个经典的树形背包问题,对于每一个边都有依赖关系,我们首先要考虑选择上面的边,因为我们选上面的边必须要选下面的边,所以
一开始递归到上层的时候就需要留空间出来
AC代码如下:
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
int edge[1000],h[1000],ne[1000];
int f[1000][1000],w[1000];
int cnt;
int n,m;
void add(int p,int x,int y)
{
w[cnt]=p,edge[cnt]=x,ne[cnt]=h[y],h[y]=cnt++;
}
int dfs(int x,int fa){
int u=0; //u为减少计算量,可以省略
for(int i=h[x];i!=-1;i=ne[i]){
int y=edge[i];
if(y==fa)continue;
u+=dfs(y,x)+1;
for(int j=min(m,u);j>=1;j--){
for(int k=1;k<=min(j,u);k++){
f[x][j]=max(f[x][j],f[x][j-k]+f[y][k-1]+w[i]);
}
}
}
return u;
}
int main()
{
cin>>n>>m;
for(int i=0;i<=n;i++)h[i]=-1;
for(int i=1;i<n;i++){
int a,b,c;
cin>>a>>b>>c;
add(c,a,b);
add(c,b,a);
}
dfs(1,0);
cout<<f[1][m]<<endl;
return 0;
}