【HAOI2015】【BZOJ4033】T1树上染色

Description
有一棵点数为 N 的树，树边有边权。给你一个在 0~ N 之内的正整
数 K ，你要在这棵树中选择 K个点，将其染成黑色，并将其他 的
N-K个点染成白色 。 将所有点染色后，你会获得黑点两两之间的距
离加上白点两两之间的距离的和的受益。问受益最大值是多少。
Input

第一行包含两个整数 N, K 。
接下来 N-1 行每行三个正整数 fr, to, dis ， 表示该树中存在一条长度
为 dis 的边 (fr, to) 。输入保证所有点之间是联通的。
Output

输出一个正整数，表示收益的最大值。
Sample Input
3 1
1 2 1
1 3 2
Sample Output
3
HINT

对于 100% 的数据， 0<=K<=N <=2000

Source

鸣谢bhiaibogf提供

f[i][j]表示i子书,j个黑点
暴力枚举son[i]的子树中黑点个数和i的子树黑点个数转移一下
容斥计算权值
看似n^3实际n^2
有个复杂度证明.懒得证了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 2010
#define LL long long
#define GET (ch>='0'&&ch<='9')
using namespace std;
int n,K,top,now;
int size[MAXN];
const long long MAXINT=1ll<<60;
LL f[MAXN][MAXN<<1],tmp[MAXN<<1];
void in(int &x)
{
    char ch=getchar();x=0;int flag=1;
    while (!GET)    flag=ch=='-'?-1:flag,ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();x*=flag;
}
struct edge
{
    int to,w;
    edge *next;
}e[MAXN<<1],*prev[MAXN];
void insert(int u,int v,int w)
{
    e[++top].to=v;e[top].next=prev[u];prev[u]=&e[top];e[top].w=w;
}
void dfs(int x,int fa)
{
    size[x]=1;
    for (int i=2;i<=n;i++)  f[x][i]=-MAXINT;
    for (edge *i=prev[x];i;i=i->next)
        if (i->to!=fa)
        {
            dfs(i->to,x);
            for (int j=0;j<=size[x]+size[i->to];j++)    tmp[j]=-MAXINT;
            for (int j=0;j<=size[x];j++)
                for (int k=0;k<=size[i->to];k++)
                    tmp[j+k]=max(tmp[j+k],f[x][j]+f[i->to][k]+(LL)i->w*(k*(K-k)+(size[i->to]-k)*(n-K-(size[i->to]-k))));
            size[x]+=size[i->to];
            for (int j=0;j<=size[x];j++)    f[x][j]=tmp[j];
        }
}
int main()
{
    in(n);in(K);
    for (int i=1;i<n;i++)
    {
        int u,v,w;in(u);in(v);in(w);
        insert(u,v,w);insert(v,u,w);
    }
    dfs(1,0);
    cout<<f[1][K]<<endl;
}