P2015

二叉苹果树

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
const int maxn=200;
struct edge{
    int to,nxt,dis;
}e[maxn];
int n,m;
inline int read()
{
    int x;char c=getchar();
    while(c<'0' or c>'9')c=getchar();
    x=c-'0',c=getchar();
    while(c>='0' and c<='9')x=x*10+c-'0',c=getchar();
    return x;
}
int head[maxn],ecnt,f[maxn][maxn];
inline void addedge(int from,int to,int dis)
{
    e[++ecnt]=(edge){to,head[from],dis},head[from]=ecnt;
}
int siz[maxn];
void dfs(int x,int fa)
{
    for(int i=head[x];i;i=e[i].nxt)
    {
        int u=e[i].to;
        if(u==fa)continue;
        dfs(u,x);
        siz[x]+=siz[u]+1;
        for(int j=min(siz[x],m);j;j--)
            for(int k=min(siz[u],j-1);k>=0;k--)
            f[x][j]=max(f[x][j],f[x][j-k-1]+f[u][k]+e[i].dis);
            //j表示保留j时，k全部遍历所有情况 
    }
}

int main()
{
    n=read(),m=read();
    for(int a,b,c,i=1;i<n;i++)
    {
        a=read(),b=read(),c=read();
        addedge(a,b,c);addedge(b,a,c);
    }
    dfs(1,0);
    printf("%d",f[1][m]);
    return 0;
}