URAL_1018 Binary Apple Tree 树形DP+背包

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 210
using namespace std;
int u[N],v[N],e[N],nt[N],ft[N];
int dp[N][N];
int cnt;
void add(int a,int b,int val)
{
    u[cnt]=a;
    v[cnt]=b;
    e[cnt]=val;
    nt[cnt]=ft[a];
    ft[a]=cnt++;
}
int n,k;
int sum[N];
void dfs(int x,int f)
{
    sum[x]=1;
    for (int i=ft[x];i>=0;i=nt[i]){
        int nx=v[i];
        if (nx==f) continue;
        dfs(nx,x);
        sum[x]+=sum[nx];
        for (int j=sum[x];j>=1;j--){
            for (int w=1;w<=sum[nx] && w<j;w++){
                dp[x][j]=max(dp[x][j],dp[x][j-w]+dp[nx][w]+e[i]);
            }
        }
    }
}
int main()
{
    while (scanf("%d%d",&n,&k)!=EOF)
    {
        cnt=0;
        int a,b,c;
        memset(ft,-1,sizeof ft);
        for (int i=1;i<n;i++){
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
            add(b,a,c);
        }
        memset(dp,0,sizeof dp);
        dfs(1,-1);
        printf("%d\n",dp[1][k+1]);
    }
    return 0;
}