[HAOI2015]bzoj4033树上染色/洛谷P3177

思路比较清晰（然而只在洛谷上过了。。。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;
}