二叉苹果数

思路：首先建树，用一个ma数组，储存输入的数据，用三个数组，l[N],r[N],a[N],完成数的建立，l[i]存储i的左子树顶点，r[i]存储右子树顶点，a[i]存储以i为终点的树枝上的苹果数。建树的过程是两个for循环加递归分别建立左子树右子树。
DP函数：
因为要保留q个树枝，即要保留q+1个顶点。所以q++。dp[i][j]代表顶点i保留j个顶点所能得到的最大苹果数。用for循环所有情况。
三个递归出口：
1.当在顶点i保存0个顶点的时候返回0(即砍掉包括i在内的子树）
2.当已经循环到树端点的时候，返回端点上的苹果数
3.当dp[fa][all]>0也就是到结点fa为止，保留all个结点的最大值
注意：ma【】【】要初始化为0，因为可能有枝上苹果为0的情况

#include<bits/stdc++.h>
using namespace std;
#define mod 1e9+7
#define N 105
#define inf 0x3f3f3f3f
const double PI = atan(1.0)*4.0;
typedef long long ll;
int ma[N][N],l[N],r[N],a[N],dp[N][N];
int n,q;
void build(int v)
{
    for(int i=1;i<=n;i++)
    {
        if(ma[v][i]>=0)
        {
            l[v]=i;
            a[i]=ma[v][i];
            ma[v][i]=ma[i][v]=-1;
            build(i);
            break;
        }
    }
    for(int i=1;i<=n;i++)
    {
        if(ma[v][i]>=0)
        {
            r[v]=i;
            a[i]=ma[v][i];
            ma[v][i]=ma[i][v]=-1;
            build(i);
            break;
        }
    }
}
int DP(int fa,int all)
{
    if(all==0)return 0;
    if(l[fa]==0&&r[fa]==0)return a[fa];
    if(dp[fa][all]>0) return dp[fa][all];
    for(int i=0;i<=all-1;i++)
    {
        dp[fa][all]=max(dp[fa][all],DP(l[fa],i)+DP(r[fa],all-1-i)+a[fa]);
    }
    return dp[fa][all];
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    freopen("E:\\in.txt","r",stdin);
for(int i=0;i<=N;i++)
    for(int j=0;j<=N;j++)
      ma[i][j]=-1;

    cin>>n>>q;
    q++;

    int x,y,z;
    for(int i=1;i<=n;i++)
    {
        cin>>x>>y>>z;
        ma[x][y]=z;
        ma[y][x]=z;
    }
    build(1);

    cout<<DP(1,q)<<endl;


}