ural 1018(ツリーdp)
7485 ワード
タイトルリンク:http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=17662
構想:典型的な木形dpは,処理時にパケットバックパックに類似し,dp[i][j]はiをルートとする木がj個の分岐を取って得た最大値を表す.
View Code
構想:典型的な木形dpは,処理時にパケットバックパックに類似し,dp[i][j]はiをルートとする木がj個の分岐を取って得た最大値を表す.
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
",dp[1][q+1]);
51 }
52 return 0;
53 }View Code