4033: [HAOI2015]T1 树型DP

第一思路是用f[i][j]表示以i为根的子树中j个黑色的最大值，然后发现根本无法转移的样子qwq。。。。
于是换一种思路。。
用f[i][j]表示以i为根的子树中j个黑色对答案的最大贡献，这样我们就可以将子树内的和子树外的进行计算了。。
具体见代码。。

#include<iostream>
#include<cstdio>
#include<cstring>
#define ll long long 
using namespace std;
int n,k,cnt,head[2005],list[4005],next[4005],key[4005],size[2005];
ll f[2005][2005];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int x,int y,int z)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
    key[cnt]=z;
}
void dp(int x,int fa)
{
    size[x]=1;
    f[x][0]=f[x][1]=0;
    for (int i=head[x];i;i=next[i])
        if (list[i]!=fa)
        {
            dp(list[i],x);
            size[x]+=size[list[i]];
            for (int j=size[x];~j;j--)
                for (int l=0;l<=min(j,size[list[i]]);l++)
                    f[x][j]=max(f[x][j],f[list[i]][l]+f[x][j-l]+((ll)(l*(k-l))+(ll)(size[list[i]]-l)*(ll)(n-k-(size[list[i]]-l)))*(ll)(key[i]));
        }
}   
int main()
{
    n=read(); k=read();
    memset(f,128,sizeof(f));
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read(),w=read();
        insert(u,v,w); insert(v,u,w);
    }
    dp(1,0);
    cout << f[1][k] << endl;
    return 0;
}