BZOJ4033：[HAOI2015]树上染色（树形dp）

题面
题意：n个点有边权的树，让你选择k个点为白色，其余为黑色。
问同色点两两之间距离的和的最大值。

设$f[x][i]$为子树x里i个白点的最大贡献
包括x的父边边权的贡献

然后一个一个儿子合并，字面意思转移

看似$n^3$，其实转移时相当于在lca处枚举点对，故是$n^2$的

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=2020;

int n,k;
int head[N],to[N<<1],nex[N<<1],val[N<<1],cnt;
int siz[N],fa[N];
LL f[N][N],by[N];

void add(int u,int v,LL w)
{
    to[++cnt]=v;
    val[cnt]=w;
    nex[cnt]=head[u];
    head[u]=cnt;
}

void dfs(int x,LL len)
{
    siz[x]=1;
    for(int h=head[x];h;h=nex[h])
    if(to[h]!=fa[x])
    {
        fa[to[h]]=x;
        dfs(to[h],val[h]);

        for(int i=0;i<=min(siz[x],k);i++)
        by[i]=f[x][i];
        for(int i=0;i<=min(siz[x],k);i++)
        for(int j=0;j<=min(siz[to[h]],k);j++)
        f[x][i+j]=max(f[x][i+j],by[i]+f[to[h]][j]);
        siz[x]+=siz[to[h]];
    }
    for(int i=0;i<=min(siz[x],k);i++)
    f[x][i]+=len*(LL)(i*(k-i)+(siz[x]-i)*(n-k-siz[x]+i));
}

int main()
{
    cin>>n>>k;

    for(int i=1;i<n;i++)
    {
        int u,v;
        LL w;
        scanf("%d%d%lld",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    dfs(1,0);
    cout<<f[1][k]<<endl;

    return 0;
}