（noip 模擬 染色）

Problem

染色
【題目描述】
有一棵點數爲 n 的樹，樹邊有邊權。將 m 個點染成黑色，並將其他的點染
成白色。會獲得黑點兩兩之間的距離和加上白點兩兩之間的距離和的收益。問
收益最大值是多少。
【輸入格式】
第一行兩個整數 n、m。接下來 n-1 行，每行三個整數 a、b、c，表示有一
條樹邊連接 a、b，長度爲 c。
【輸出格式】
一行一個正整數，表示收益的最大值。
【樣例輸入】
3 1
1 2 1
1 3 2
【樣例輸出】
3
【數據規模與約定】
30%的數據，n≤15
60%的數據，n≤100
另 20%的數據，這棵樹是一條鏈
100%的數據，0≤m≤n≤2000，1≤c≤1,000,000

有點像揹包

Code

// by spli
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define LL long long
using namespace std;

const int M=2010;
int n,m;
struct node{
    int to,nxt;
    LL val;
}e[M<<1];int head[M],cnt,siz[M];
int fa[M];
LL f[M][M];

void dfs(int u){
    for(int i=head[u];i!=-1;i=e[i].nxt){
        int v=e[i].to;
        if(v!=fa[u]){
            fa[v]=u;
            dfs(v);
            for(int a=siz[u];a>=0;--a){
                for(int b=siz[v];b>=0;--b){
                    f[u][a+b]=max(f[u][a+b],f[u][a]+f[v][b]+b*(m-b)*e[i].val+(siz[v]-b)*(n-siz[v]-m+b)*e[i].val);       
                }
            }
        }
    }
}

void add(int f,int t,int v){
    cnt++;
    e[cnt]=(node){t,head[f],v};
    head[f]=cnt;
    siz[f]++;
}

int main(){
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    int x,y,z;
    for(int i=1;i<=n;++i) siz[i]++;
    for(int i=1;i<n;++i){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);add(y,x,z);
    }
    dfs(1);
        cout<<endl;
    }*/
    cout<<f[1][m];
    return 0;
}