【树型DP】二叉苹果树--------------------------------------dp

第二天叫醒我的不是闹钟，是梦想！

题目描述
有一棵二叉苹果树，如果数字有分叉，一定是分两叉，即没有只有一个儿子的节点。这棵树共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;
}