BZOJ4033 HAOI2015 树上染色

又是小朋友们【大雾】的题

所以不是很难啦

简单的边权贡献树形dp

边权没给 要开longlong

emm具体细节什么的可以看代码w

唯一的细节可能就是第一重循环要倒着 因为要取w-j更新

然后双向边...数组记得开两倍（

//Love and Freedom.
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#define ll long long
#define inf 20021225
#define N 2001
using namespace std;
struct edg{int to,lt,v;}e[N<<1];
ll f[N][N]; int cnt,in[N],k,sz[N],n;
void add(int x,int y,int v)
{
    e[++cnt].to=y; e[cnt].lt=in[x]; e[cnt].v=v; in[x]=cnt;
    e[++cnt].to=x; e[cnt].lt=in[y]; e[cnt].v=v; in[y]=cnt;
}
void dfs(int x,int fa)
{
    sz[x]=1; memset(f[x],-1,sizeof(f[x])); f[x][0]=f[x][1]=0;
    for(int i=in[x];i;i=e[i].lt)
    {
        int y=e[i].to,v=e[i].v; if(y==fa)    continue;
        dfs(y,x); sz[x]+=sz[y]; int top=min(k,sz[x]);
        for(int w=top;~w;w--)
            for(int j=0;j<=min(w,sz[y]);j++)
            {
                if(f[x][w-j]==-1)    continue;
                f[x][w]=max(f[x][w],f[x][w-j]+f[y][j]+1ll*v*(sz[y]-j)*(n-sz[y]-k+j)+1ll*v*j*(k-j));
            }
    }
}
int main()
{
    scanf("%d%d",&n,&k); int x,y,v;
    for(int i=1;i<n;i++)
        scanf("%d%d%d",&x,&y,&v),add(x,y,v);
    dfs(1,1);
    printf("%lld\n",f[1][k]);
    return 0;
}