树型DP（分组背包问题）：二叉苹果树

题目链接：http://ybt.ssoier.cn:8088/statusx.php?runidx=11423869

题目：

【题目描述】 有一棵二叉苹果树，如果数字有分叉，一定是分两叉，即没有只有一个儿子的节点。这棵树共 N 个节点，标号 11 至 NN ，树根编号一定为 1 。 我们用一根树枝两端连接的节点编号描述一根树枝的位置。一棵有四根树枝的苹果树，因为树枝太多了，需要剪枝。但是一些树枝上长有苹果，给定需要保留的树枝数量，求最多能留住多少苹果。     【输入】 第一行两个数 N 和 Q ，N 表示树的节点数，Q 表示要保留的树枝数量。 接下来 N−1 行描述树枝信息，每行三个整数，前两个是它连接的节点的编号，第三个数是这根树枝上苹果数量。 【输出】 输出仅一行，表示最多能留住的苹果的数量。 【输入样例】 5 2 1 3 1 1 4 10 2 3 20 3 5 20 【输出样例】 21 【提示】 数据范围与提示： 对于 100% 的数据，1≤Q≤N≤100,N≠1，每根树枝上苹果不超过 30000 个。

分析：

 首先此题已经说明，是必须要与1相连通的m条边。

有n个节点，n - 1条边， 这道题目就可以看作，n组数（节点），每个节点作为根节点，其子节点对应的边数会有k条，则k就是每组数可以选择的方案，选择1条，则为某个节点作为根节点，与它相邻的一条边，  选择2条，则为某个节点作为根节点，与它相邻的两条边，或者相通的两条边。以此类推。  总背包容量为m。

f[i][j] , 状态表示：以 i节点为根节点，选择与i节点相通的j条边的最大值 (此j条边为多个子节点中的分配一些边，总共j条边的情况。)。

状态计算：

i 对应的节点有：

for(int k = h[i] ; k != -1 ; k = ne[k])

{

        int j = e[k];

}

对应的节点j就是其子节点。

f[i][temp] = max(  f[i][temp - k - 1] + f[j][k] + w[] )

要注意：此f[i][temp - k - 1]的情况下，是遍历到了i节点的前多个，目前最优情况下是f[i][temp - k -1]的情况，给j分k条边。

举例分析（以多叉树为例）：

 void dfs(int a , int father)
{
    for(int i = h[a] ; i != -1 ; i = ne[i])
    {
        int t = e[i];
        if(t == father)
        {
            continue;
        }
        dfs(t , a);  // 先算出子节点的f[][]，在用子节点的f[][]算出根节点的f[][]
        for(int j = m ; j >= 0 ;j--)
        {
            for(int k = 0 ; k < j ; k++)
            {
                f[a][j] = max(f[a][j],f[t][k] + f[a][j - k - 1] + w[i]);  // t作为根节点，选择k条子边，那么i作为根节点，除了t的那k条，外加一条i->t的边，其余选择j - k - 1条
            }
        }
    }
}

其中的f[i][temp] = max(  f[i][temp - k - 1] + f[j][k] + w[] )  

当遍历到了j这个节点，f[i][temp - k - 1]为已经分好了u,p共temp - k - 1条边情况下的最大值，然后给j了k条边。共temp条边，但是为分给了u,p,j个不同边后的最大值情况。然后在遍历最后一个子节点看最后一个子节点分多少条边 “？” 。u,p,j共分temp - "?" - 1条边. 为什么 - 1。这个1的含义实际上就对应的是 w[]，

从而通过dfs()的方式，不断的将所有子节点的f[][]的所有情况下最优的结果存入f[][]中，最后到根节点1。算出根节点1的f[][]即可。

最后输出f[1][m].

 代码实现：

# include <iostream>
# include <cstring>
# include <stdio.h>
using namespace std;

const int N  = 110 , M = 2 * N;

int f[N][N];  // f[i][j]的含义是：以i为根节点，选择i的子节点中的j条边

int n,m;

int h[N],e[M],ne[M],w[M],idx;

void add(int a , int b , int wei)
{
    e[idx] = b;
    w[idx] = wei;
    ne[idx] = h[a];
    h[a] = idx++;
}

void dfs(int a , int father)
{
    for(int i = h[a] ; i != -1 ; i = ne[i])
    {
        int t = e[i];
        if(t == father)
        {
            continue;
        }
        dfs(t , a);  // 先算出子节点的f[][]，在用子节点的f[][]算出根节点的f[][]
        for(int j = m ; j >= 0 ;j--)
        {
            for(int k = 0 ; k < j ; k++)
            {
                f[a][j] = max(f[a][j],f[t][k] + f[a][j - k - 1] + w[i]);  // t作为根节点，选择k条子边，那么i作为根节点，除了t的那k条，外加一条i->t的边，其余选择j - k - 1条
            }
        }
    }
}

int main()
{
    memset(h,-1,sizeof h);
    scanf("%d %d",&n,&m);
    for(int i = 0 ; i < n - 1 ; i++)
    {
        int a,b,wei;
        scanf("%d %d %d",&a,&b,&wei);
        add(a,b,wei);
        add(b,a,wei);
    }

    dfs(1 , - 1);
    printf("%d\n",f[1][m]);
    return 0;
}