ural 1018(树形dp)

1 #include<iostream>
2 #include<cstdio>
3 #include<cstring>
4 #include<algorithm>
5 #include<vector>
6 using namespace std;
7 #define MAXN 111
8
9 struct Edge{
10     int v,w;
11     Edge(){}
12     Edge(int vv,int ww):v(vv),w(ww){}
13 };
14
15 int n,q;
16 int num[MAXN];
17 int dp[MAXN][MAXN];//以i为根的树选了j个分支（包括i）
18
19 vector<vector<Edge> >g;
20 void dfs(int u,int father)
21 {
22     num[u]=1;
23     for(int i=0;i<g[u].size();i++){
24         int v=g[u][i].v,w=g[u][i].w;
25         if(v==father)continue;
26         dfs(v,u);
27         num[u]+=num[v];
28         for(int j=num[u];j>=1;j--){
29             for(int k=1;k<j&&k<=num[v];k++){
30                 dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]+w);
31             }
32         }
33     }
34 }
35
36
37 int main()
38 {
39     int u,v,w;
40     while(~scanf("%d%d",&n,&q)){
41         g.clear();
42         g.resize(n+2);
43         for(int i=1;i<n;i++){
44             scanf("%d%d%d",&u,&v,&w);
45             g[u].push_back(Edge(v,w));
46             g[v].push_back(Edge(u,w));
47         }
48         memset(dp,0,sizeof(dp));
49         dfs(1,-1);
50         printf("%d\n",dp[1][q+1]);
51     }
52     return 0;
53 }