ural 1018(树形dp)

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

思路：典型的树形dp,处理的时候类似于分组背包，dp[i][j]代表以i为根的树取j个分支获得的最大值。

#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
",dp[1][q+1]);
    }
    return 0;
}