二叉苹果树

出题人：赵老师

描述

有一棵二叉苹果树，如果数字有分叉，一定是分两叉，即没有只有一个儿子的节点。这棵树共 N 个节点，标号 1 至 N，树根编号一定为 1。

我们用一根树枝两端连接的节点编号描述一根树枝的位置。一棵有四根树枝的苹果树，因为树枝太多了，需要剪枝。但是一些树枝上长有苹果，给定需要保留的树枝数量，求最多能留住多少苹果。

输入描述

第一行两个数 N 和 Q ，N 表示树的节点数，Q 表示要保留的树枝数量。

接下来 N−1 行描述树枝信息，每行三个整数，前两个是它连接的节点的编号，第三个数是这根树枝上苹果数量。

输出描述

输出仅一行，表示最多能留住的苹果的数量。

用例输入 1 

5 2
1 3 1
1 4 10
2 3 20
3 5 20

用例输出 1 

21

提示

数据范围与提示：

对于 100% 的数据，1≤Q≤N≤100,N=1，每根树枝上苹果不超过 30000 个

AC

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#include <string>
#include <queue>
#include <map>
/* #include <unordered_map> */
#include <bitset>
#include <vector>
void fre() { system("clear"), freopen("A.txt", "r", stdin); freopen("Ans.txt","w",stdout); }
void Fre() { system("clear"), freopen("A.txt", "r", stdin);}
#define ios ios::sync_with_stdio(false)
#define Pi acos(-1)
#define pb push_back
#define fi first
#define se second
#define ll long long
#define ull unsigned long long
#define db double
#define Pir pair<int, int>
#define m_p make_pair
#define INF 0x3f3f3f3f
#define esp 1e-7
#define for_(i, s, e) for(int i = (ll)(s); i <= (ll)(e); i ++)
#define rep_(i, e, s) for(int i = (ll)(e); i >= (ll)(s); i --)
#define sc scanf
#define pr printf
#define sd(a) scanf("%d", &a)
#define ss(a) scanf("%s", a)
#define size() size() * 1LL
#define mod (ll)(998244353)
#define Min(a, b, c) min(a, min(b, c))
using namespace std;

const int mxn = 105;
int dp[mxn][mxn];
vector<int> e[mxn];
int w[mxn][mxn];
int n, m;

void dfs(int u, int p)
{
    if(! e[u].size()) return;
    for_(i, 0, e[u].size() - 1)
    {
        int v = e[u][i];
        if(v == p) continue;
        dfs(v, u);

        //这里的两重for循环dp枚举思路就像 01背包的动态规划，我们当前在节点u，如果我们给子节点v（把u的任意一个子节点v，在不同空间下可以看作不同的物品）分配b条边，在背包总空间为a的基础上,又因为边u-v 也占用一条边（一个空间），所以剩下的边数（空间）为 a - b - 1，
        rep_(a, m, 1)               //从大到小枚举，当前可以分配的总的边数（其实就像 从大到下枚举背包的空间，防止重复购买某个物品） 
            for_(b, 0, a - 1)
                dp[u][a] = max(dp[u][a], dp[v][b] + dp[u][a - b - 1] + w[u][v]);
    }
}


int main()
{
    /* fre(); */
    sc("%d %d", &n, &m);
    int u, v, x;
    for_(i, 2, n)
    {
        sc("%d %d %d", &u, &v, &x);
        e[u].pb(v);
        e[v].pb(u);
        w[u][v] = x;
        w[v][u] = x;
    }

    dfs(1, 0);
    pr("%d\n", dp[1][m]);

    return 0;
}