P2015 二叉苹果树

# 二叉苹果树

## 题目描述

有一棵苹果树，如果树枝有分叉，一定是分二叉（就是说没有只有一个儿子的结点）

这棵树共有 N个结点（叶子点或者树枝分叉点），编号为 1 ~ N，树根编号一定是 1。

我们用一根树枝两端连接的结点的编号来描述一根树枝的位置。下面是一颗有 4 个树枝的树：

```
2   5
 \ / 
  3   4
   \ /
    1
```

现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。

给定需要保留的树枝数量，求出最多能留住多少苹果。

## 输入格式

第一行 2个整数 N 和 Q，分别表示表示树的结点数，和要保留的树枝数量。 

接下来 N-1 行，每行 3 个整数，描述一根树枝的信息：前2个数是它连接的结点的编号，第 3 个数是这根树枝上苹果的数量。

## 输出格式

一个数，最多能留住的苹果的数量。

## 样例 #1

### 样例输入 #1

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

### 样例输出 #1

```
21
```

#include<iostream>
using namespace std;
const int maxn = 110;
struct node
{
    int to, ne, w;
}edge[maxn * maxn];
int head[maxn];
int cnt;
int in[maxn];
int dp[maxn][maxn];
int n, q;
void add(int u,int v,int w)
{
    edge[++cnt].ne = head[u];
    edge[cnt].w = w;
    edge[cnt].to = v;
    head[u] = cnt;
}
void dfs(int u)
{
    for(int i = head[u]; i; i = edge[i].ne)
    {
        int v = edge[i].to;
        dfs(v);
        for(int j = q; j >= 1; j--)
        {
            for(int k = 1; k <= j; k++)
            {
                dp[u][j] = max(dp[u][j], dp[u][j - k] + dp[v][k - 1] + edge[i].w);
            }
        }
    }
}
int main()
{
    cin >> n >> q;
    int x, y, w;
    for(int i = 1; i < n; i++)
    {
        cin >> x >> y >> w;
        add(x, y, w);
        in[y]++;
    }
    int root;
    for(int i =1 ;i <= n; i++)
    {
        if(!in[i])
        {
            root = i;
            break;
        }
    }
    dfs(root);
    cout << dp[root][q] <<endl;
    return 0;
}