二叉苹果树

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=107;
int map[maxn][maxn],l[maxn],r[maxn],a[maxn],f[maxn][maxn];
int n,q;
void buildtree(int u){
    for(int i=1;i<=n;i++){
        if(map[u][i]!=-1){
            l[u]=i;a[i]=map[u][i];
            map[u][i]=-1;map[i][u]=-1;
            buildtree(i);
            break;
        }
    }
    for(int i=1;i<=n;i++){
        if(map[u][i]!=-1){
            r[u]=i;a[i]=map[u][i];
            map[u][i]=-1;map[i][u]=-1;
            buildtree(i);
            break;
        }
    }
}
int DP(int i,int j){
    if(j==0) return f[i][j]=0;
    if(l[i]==0&&r[i]==0) return f[i][j]=a[i];
    if(f[i][j]>0) return f[i][j];
    for(int k=0;k<=j-1;k++)
        f[i][j]=max(f[i][j],DP(l[i],k)+DP(r[i],j-k-1)+a[i]);
    return f[i][j];
} 
int main(){
    cin>>n>>q;
    memset(map,-1,sizeof(map));
    for(int i=1;i<n;i++){
        int u,v,w;cin>>u>>v>>w;
        map[u][v]=w;map[v][u]=w;
    } 
    buildtree(1);
    int ans=DP(1,q+1);
    cout<<ans<<endl;
    return 0;
}