p3177 [HAOI2015]树上染色

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<ctime>
#include<vector>
#include<set>
#include<map>
#include<stack>
using namespace std;
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define int long long
vector<pair<int,int> >v[2010];
int dp[2010][2010],siz[2010],n,k;
inline void dfs(int x,int fa){
    siz[x]=1;
    for(int _=0;_<v[x].size();_++)
      if(v[x][_].fi!=fa){
          int y=v[x][_].fi,z=v[x][_].se;
          dfs(y,x);
          siz[x]+=siz[y];
      }
    memset(dp[x],-1,sizeof(dp[x]));
    dp[x][0]=dp[x][1]=0;
    for(int _=0;_<v[x].size();_++)
      if(v[x][_].fi!=fa){
          int y=v[x][_].fi,z=v[x][_].se;
          for(int i=min(siz[x],k);i>=0;i--)
            for(int j=0;j<=min(siz[y],i);j++){
                if(dp[x][i-j]==-1)continue; 
                int res=j*(k-j)*z+(siz[y]-j)*(n-k-siz[y]+j)*z;
                dp[x][i]=max(dp[x][i],dp[x][i-j]+dp[y][j]+res);
            }
      }
    return;
}
signed main(){
    int i,j;
    scanf("%lld%lld",&n,&k);
    for(i=1;i<n;i++){
      int x,y,z;
      scanf("%lld%lld%lld",&x,&y,&z);
      v[x].pb(mp(y,z));
      v[y].pb(mp(x,z));
    }
    dfs(1,0);
    printf("%lld\n",dp[1][k]);
    return 0;
}