洛谷 二叉苹果树（树形dp，类01背包解决（可解决多叉树））

 题目链接：

二叉苹果树 - 洛谷

思路：

f[i][j]表示i号结点，保留j个树枝，最多的苹果数。

每次更新一个子结点，然后计算当前的 f 值，直到遍历完所有子结点，得到最终的 f 值。

类似01背包，所以注意 i 必须倒序遍历。详见代码。

代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 105;
int n, m, size[maxn]; //size表示子树大小
int f[maxn][maxn]; // f[i][j]表示i号结点，保留j个树枝，最多的苹果数
struct node{int v,w;};
vector<node> g[maxn];

void dfs(int u,int fa){
    size[u] = 1; //初始大小是1

    //每次只更新一个子结点，类似01背包的思路，结点代表物品，树枝数代表背包容量
    //当然，实际取物品比01背包复杂，所以还要加一个j的for循环用来遍历当前子结点的子树
    for(auto P : g[u]){
        int v=P.v, w=P.w;
        if(v==fa) continue;
        dfs(v,u);
        size[u] += size[v];

        //特别注意这里的遍历顺序，i必须倒序遍历，j正序倒序都可
        //i倒序的原因：这里递推用的f[u][i-j-1]是之前的子结点对应的值，要保证该f值在本层没被更新过
        //类似01背包，每层更新一个新结点，最后汇总所有结点的f值
        //而j表示在当前子结点取几个树枝，这个遍历顺序没有关系
        for(int i=min(size[u]-1,m); i; i--) //遍历当前要求的保留树枝数量
        for(int j=min(size[v]-1, i-1); j>=0; j--) //注意j可以取到0，表示要u->v这条边，但是不要v的子边
            f[u][i] = max(f[u][i], f[u][i-j-1]+f[v][j]+w);
    }
}
int main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin>>n>>m; //结点数，保留树枝数
    for(int i=1; i<=n-1; i++){
        int a, b,c; cin>>a>>b>>c;
        g[a].push_back({b,c});
        g[b].push_back({a,c});
    }
    dfs(1,1);
    cout<<f[1][m];
}