P2015 二叉苹果树（树形dp）

#include<iostream>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 1000;
int n, m, h[N], cnt, f[N][N], sz[N];
struct node {
    int v, w, net;
} no[N];
void add(int u, int v, int w) {
    no[cnt] = {v, w, h[u]};
    h[u] = cnt++;
}
void dfs(int u, int fa) {
    for(int i = h[u]; ~i; i = no[i].net) {
        int v = no[i].v;加的双向边，如果下一个是父节点，说明已经递归到底了，递归父节点的兄弟节点
        if(v == fa)
            continue;
        dfs(v, u);
        sz[u] += sz[v] + 1;//sz[i]表示i节点的子节点数
        for(int j = min(m, sz[u]); j; j--)//背包
            for(int k = min(j - 1, sz[v]); k >= 0; k--)
                f[u][j] = max(f[u][j], f[u][j - k - 1] + f[v][k] + no[i].w);//j-k-1是还有一个父节点与它连的边
    }
}
int main() {
    memset(h, -1, sizeof h);
    cin >> n >> m;
    for(int u, v, w, i = 1; i < n; i++) {
        cin >> u >> v >> w;
        add(u, v, w);
        add(v, u, w);
    }
    dfs(1, -1);
    cout << f[1][m];
    return 0;
}