P2015 二叉苹果树（树形DP）

题目描述
有一棵苹果树，如果树枝有分叉，一定是分2叉（就是说没有只有1个儿子的结点）

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

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

2 5
\ /
3 4
\ /
1
现在这颗树枝条太多了，需要剪枝。但是一些树枝上长有苹果。

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

输入格式
第1行2个数，N和Q(1<=Q<= N,1<N<=100)。

N表示树的结点数，Q表示要保留的树枝数量。接下来N-1行描述树枝的信息。

每行3个整数，前两个是它连接的结点的编号。第3个数是这根树枝上苹果的数量。

每根树枝上的苹果不超过30000个。

输出格式
一个数，最多能留住的苹果的数量。

输入输出样例
输入 #1 复制
5 2
1 3 1
1 4 10
2 3 20
3 5 20
输出 #1 复制
21

刚做完上司那道题，原以为树形dp还挺简单，做完这一题就懵了，都说是一道很裸的dp，我他妈愣是看了两个小时，智商感人；
首先dp开一个二维数组，dp[i][j]代表第i个结点，保留j条边最多能剩多少个苹果；首先要求剩下q条边，这是背包的容量，每一个父节点的状态都是子节点转移过来的；
转移方程为：

具体解释见代码：

#include <bits/stdc++.h>

using namespace std;
const int maxx=5000;
int n,q;
int dp[maxx][maxx];
int e[maxx],ne[maxx],h[maxx],id;
int dis[maxx];
void add(int a,int b,int c)//这里我用邻接表存图
{
    dis[id]=c,e[id]=b,ne[id]=h[a],h[a]=id++;
}
int dfs(int x,int y)//x为子节点，y为父节点
{
    int d=0;
    for(int i=h[x];i!=-1;i=ne[i])
    {
        int ans=e[i],cot=dis[i];//与x点相连的结点为ans，权值为cot
        if(y==ans) continue;//当此节点的父节点和原来相等时，那么朔门到了最底层；
        //d+=dfs(ans,x)+1;
        dfs(ans,x);//首先当然递归到最底层，然后不断返回状态；
        for(int j=q;j>=0;j--)//背包的容量是q条边
            for(int k=j-1;k>=0;k--)//每个子树的父节点都可以去点小于等于j-1个结点
                dp[x][j]=max(dp[x][j],dp[x][j-k-1]+dp[ans][k]+cot);
    }
    return d;
}
int main()
{
    cin >>n>>q;
    for(int i=0;i<=n;i++) h[i]=-1;
    for(int i=1;i<n;i++)
    {
        int a,b,c;
        cin >>a>>b>>c;
        add(a,b,c);
        add(b,a,c);//无向图所以要存两边！
    }
    dfs(1,0);
    cout <<dp[1][q]<<endl;//输出结点1保留q条边最多的苹果数
    return 0;
}