思路比较清晰(然而只在洛谷上过了。。。bzoj不知道为何一直TLE),然而状态转移方程可以说是十分复杂了。。。。
我们定义一个dp数组dp[u][k],表示以u这个节点为根节点,在它的子树中寻找k个节点染色而可以获得的最大值,然后用总数K减去u为跟的子树内部染的k个节点,来算出节点外的值,然后更新dp数组。
状态转移方程大概如下
对于任意一个节点u,在它的内部染色k个节点,整棵树上可以染的的节点总数为K,v为u的子节点。
dp[u][j]=max(dp[u][j],dp[u][j-k]+dp[v][k]+((k*(K-k))+(siz[v]-k)*(N-K-(siz[v]-k))*edge[i].w));
所以说代码的实现除了dp的部分之外还是轻松愉快的
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=2005;
typedef long long LL;
int a[MAXN];
int fat[MAXN];
struct Edge
{
int nxt;
int to;
int w;
}edge[MAXN<<1];
int head[MAXN<<1];
int N,K;
LL dp[MAXN][MAXN];
int siz[MAXN];
int num;
void add(int from,int to,int w)
{
edge[++num].nxt=head[from];
edge[num].to=to;
edge[num].w=w;
head[from]=num;
}
void dfs(int u,int pre)
{
siz[u]=1;
dp[u][0]=dp[u][1]=0;
for(int i=head[u];i;i=edge[i].nxt)
{
int v=edge[i].to;
if(v==pre) continue;
dfs(v,u);
siz[u]+=siz[v];
for(int j= min(siz[u],K);j>=0;j--)
{
for(int k=0;k<= min(siz[v],j);k++)
{
dp[u][j]=std::max(dp[u][j],dp[u][j-k]+dp[v][k]+LL(((k*(K-k))+(siz[v]-k)*(N-K-(siz[v]-k)))*edge[i].w));
}
}
}
}
int main()
{
scanf("%d%d",&N,&K);
memset(dp,128,sizeof(dp));
for(int i=1;i<N;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
dfs(1,0);
printf("%lld\n",dp[1][K]);
return 0;
}