【bzoj4033】HAOI2015树上染色

树形dp。

#include<bits/stdc++.h>
#define N 2010
using namespace std;
typedef long long ll;
int n,k,tot=,head[N];
struct Node{int u,v,w,next;}G[N<<];
inline void addedge(int u,int v,int w){
    G[tot].u=u;G[tot].v=v;G[tot].w=w;G[tot].next=head[u];head[u]=tot++;
    G[tot].u=v;G[tot].v=u;G[tot].w=w;G[tot].next=head[v];head[v]=tot++;
}
ll dp[N][N];int size[N];
void dfs(int u,int fa){
    size[u]=;memset(dp[u],-,sizeof(dp[u]));dp[u][]=dp[u][]=;
    for(int i=head[u];~i;i=G[i].next){
        int v=G[i].v;if(v==fa)continue;
        dfs(v,u);size[u]+=size[v];
    }
    for(int i=head[u];~i;i=G[i].next){
        int v=G[i].v;if(v==fa)continue;int w=G[i].w;
        for(int i=min(k,size[u]);i>=;i--)
        for(int j=;j<=min(i,size[v]);j++)if(~dp[u][i-j]){
            ll val=(ll)j*w*(k-j)+(ll)(size[v]-j)*(n-k+j-size[v])*w;
            dp[u][i]=max(dp[u][i],dp[u][i-j]+dp[v][j]+val);
        }
    }
}
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
int main(){
    memset(head,-,sizeof(head));
    n=read();k=read();
    for(int i=;i<n;i++){
        int u=read(),v=read(),w=read();
        addedge(u,v,w);
    }
    dfs(,);
    printf("%lld\n",dp[][k]);
}