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 }
