记录编号	160448	评测结果	AAAAAAAAAA
题目名称	[HAOI 2015]树上染色	最终得分	100
用户昵称	cstdio	是否通过	通过
代码语言	C++	运行时间	3.102 s
提交时间	2015-04-27 15:24:25	内存使用	31.17 MiB

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
typedef long long LL;
const LL INF=1e16;
const int SIZEN=2010;
int N,K;
LL F[SIZEN][SIZEN];
vector<pair<int,int> > c[SIZEN];
int subsz[SIZEN];
void DFS(int x,int fa){
	F[x][0]=F[x][1]=0;
	subsz[x]=1;
	for(int i=0;i<c[x].size();i++){
		int u=c[x][i].first;
		if(u==fa) continue;
		DFS(u,x);
		subsz[x]+=subsz[u];
		for(int j=subsz[x];j>=0;j--){
			for(int k=0;k<=subsz[u]&&k<=j;k++){
				LL add=(LL)k*(K-k)+(LL)(subsz[u]-k)*(N-K-(subsz[u]-k));
				add*=c[x][i].second;
				add+=F[u][k];
				//cout<<x<<" "<<u<<" "<<k<<" "<<add<<endl;
				F[x][j]=max(F[x][j],F[x][j-k]+add);
			}
		}
	}
	//cout<<x<<endl;for(int i=0;i<=K;i++) cout<<F[x][i]<<" ";cout<<endl;
}
void read(void){
	scanf("%d%d",&N,&K);
	int a,b,w;
	for(int i=1;i<N;i++){
		scanf("%d%d%d",&a,&b,&w);
		c[a].push_back(make_pair(b,w));
		c[b].push_back(make_pair(a,w));
	}
}
int main(){
	freopen("haoi2015_t1.in","r",stdin);
	freopen("haoi2015_t1.out","w",stdout);
	read();
	for(int i=0;i<SIZEN;i++) for(int j=0;j<SIZEN;j++) F[i][j]=-INF;
	DFS(1,0);
	cout<<F[1][K]<<endl;
	return 0;
}