ural 1018(树形dp)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 111

struct Edge{
int v,w;
Edge(){}
Edge(int vv,int ww):v(vv),w(ww){}
};

int n,q;
int num[MAXN];
int dp[MAXN][MAXN];//以i为根的树选了j个分支（包括i）

vector<vector<Edge> >g;
void dfs(int u,int father)
{
num[u]=1;
for(int i=0;i<g[u].size();i++){
int v=g[u][i].v,w=g[u][i].w;
if(v==father)continue;
dfs(v,u);
num[u]+=num[v];
for(int j=num[u];j>=1;j--){
for(int k=1;k<j&&k<=num[v];k++){
dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]+w);
}
}
}
}

int main()
{
int u,v,w;
while(~scanf("%d%d",&n,&q)){
g.clear();
g.resize(n+2);
for(int i=1;i<n;i++){
scanf("%d%d%d",&u,&v,&w);
g[u].push_back(Edge(v,w));
g[v].push_back(Edge(u,w));
}
memset(dp,0,sizeof(dp));
dfs(1,-1);
printf("%d\n",dp[1][q+1]);
}
return 0;
}