P2015 二叉苹果树

//#define local
#include<cstdio>
#include<iostream>
#include<vector>
#define MAXN 105
using namespace std;
typedef long long ll; 
int n,q;//n结点数 q保留数 
int dp[MAXN][MAXN]={0};//dp[i][j]表示i结点后含 j个树枝的最大果实数量 
vector<int>e[MAXN];
int w[MAXN][MAXN]={0};//e[i][j]表示ij结点之间的权值 
void dfs(int cur,int fa){
	if(e[cur].size()==1){
	     return;
	}
	else{
		//cout<<"cur:"<<cur<<" "<<"fa:"<<fa<<endl;
		int r,l;
		bool cheak=false;
		for(int i=0;i<e[cur].size();++i){
			int next=e[cur][i];
			
			if(next!=fa){
				if(!cheak){
					l=next;
					cheak=true;
					
					dfs(l,cur);
				}
				else{
					r=next;
					
					dfs(r,cur);
				}
			}
		}
	
		for(int i=0;i<=q-1;++i){
		if(dp[l][i]==-1)
		break;
		dp[cur][i+1]=max(dp[cur][i+1],dp[l][i]+w[cur][l]);
		
	    }
		for(int j=0;j<=q-1;++j){
		if(dp[r][j]==-1)
		break;
		dp[cur][j+1]=max(dp[cur][j+1],dp[r][j]+w[cur][r]);
        }
		for(int i=0;i<=q-2;++i){
			if(dp[l][i]==-1)
		    break;
			for(int j=0;j<=q-i-2;++j){
			if(dp[r][j]==-1)
		    break;
			dp[cur][i+j+2]=max(dp[cur][i+j+2],dp[l][i]+dp[r][j]+w[cur][l]+w[cur][r]);
			}
		}
	}
}
int main(){
#ifdef local
freopen("data.in","rb",stdin);
//freopen("data.out","wb",stdout);
#endif
scanf("%d%d",&n,&q);
fill(dp[0],dp[0]+MAXN*MAXN,-1);
for(int i=1;i<=n;++i)
dp[i][0]=0;
for(int i=1;i<=n-1;++i){
	int u,v,g;	
	scanf("%d%d%d",&u,&v,&g);
	e[u].push_back(v);
	e[v].push_back(u); 
	w[u][v]=g;
	w[v][u]=g;
}

dfs(1,0);


printf("%d",dp[1][q]);
return 0;
}