ural 1018(树形dp)

题目链接：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=17662

思路：典型的树形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\n",dp[1][q+1]);

51     }

52     return 0;

53 }

View Code