【树形DP例题】——树上染色

---

题目

描述

有一棵点数为 n的树，树边有边权。给你一个在 0~n 之内的正整数 k，你要在这棵树中选择 k个点，将其染成黑色，并将其他的 n-k个点染成白色。将所有点染色后，你会获得黑点两两之间的距离加上白点两两之间的距离的和的收益。问收益最大值是多少。0<=n，k<=2000

输入格式

• 第一行包含两个整数n, k。
• 第二到 n行每行三个正整数u,v,w，表示该树中存在一条长度为 w 的边(u,v)。输入保证所有点之间是联通的。

输出格式

• 输出一个正整数，表示收益的最大值。

输入/输出例子

输入：
3 1
1 2 1
1 3 2
输出：
3

提示

无

---

解题思路

分析

算点两两之间的距离的和

1. 我们先不考虑颜色问题，对于这个链：

点两两之间的距离的和=5+6+（5+6）
然后我们发现图中有三条路径，分别为1->2,1->3,2->3,它们的长度分别为
5,(5+6),6，它们的和就是点两两之间的距离的和。
所以我们可以得出 点两两之间的距离的和=树中所有路径的长度和

2. 再看此图：
图中的树三条路径分别为1->2,1->3,1->2
其中，1>2经过边（1,2），1->3经过边（1,2）和（2,3），2->3经过边（2,3）
及边（1,2）和边（2,3）各经过两次，所以距离和=52+62=5+6+（5+6）
所以 树中所有路径的长度和=每条边*每条边的经过次数的和

3. 如图：

有一棵树，求树中所有路径经过蓝色边的次数。
我们通过枚举可以得出，有6条路径：1->4，1->5，2->4，2->4，3->4，3->5
6=32=边左边的点数边右边的点数
所以我们可以得出 每条边的经过次数=边左边的点数*边右边的点数

4. 这时，我们再加上颜色，那还是可以得出：
每条边的经过次数=边左边白点的点数x边右边白点的点数+边左边黑点的点数x边右边黑点的点数

动态转移方程

int sum=(k*(m-k)+(d[v]-k)*(n-m-d[v]+k))*edge[i].w;
f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+sum);
//其中f[i][j]表示以i为根节点的子树选j个节点染色后能获得的最大收益
//d[u]表示以u为根节点的子树的节点个数
//sum表示这条边的经过次数*比权

code

#include<bits/stdc++.h>
#define mod 1
#define int long long
#define foru(i,a,b) for(register int i=a;i<=b;i++)
#define ford(i,a,b) for(register int i=a;i>=b;i--)
using namespace std;
const int N=2e3+1;
int idx,head[N],n,m,d[N],f[N][N],x,y,z;
struct fy
{
	int w,to,next;
}edge[N<<1];
char gc()
{
  	static char now[1<<20],*S,*T;
  	if(T==S)
  	{
		T=(S=now)+fread(now,1,1<<20,stdin);
    	if(T==S) 
			return EOF;
  	}
  return *S++;
}
template <typename T>
void Read(T&x) 
{
  	x=0;
  	char c=gc();
  	while(c<'0'||c>'9') 
  		c=gc();
  	x=c-'0';
  	while((c=gc())>='0'&&c<='9') 
  		x=x*10+c-'0';
}
template <typename T, typename... Args>
void Read(T&x,Args&...args)
{
  	Read(x);
  	Read(args...);
}
void add(int u,int v,int w)
{
	edge[++idx].w=w,edge[idx].to=v,edge[idx].next=head[u];head[u]=idx;
}
void dfs(int u,int fa)
{
	d[u]=1;
	f[u][0]=f[u][1]=0;
	for(int i=head[u];i!=-1;i=edge[i].next)
	{
		int v=edge[i].to;
		if(v==fa)
			continue;
		dfs(v,u);
		d[u]+=d[v];
		ford(j,min(m,d[u]),0)
		{
			if(f[u][j]!=-1) 
				f[u][j]+=f[v][0]+d[v]*(n-m-d[v])*edge[i].w;
			ford(k,min(j,d[v]),1)
				if(f[u][j-k]!=-1)
				{
					int sum=(k*(m-k)+(d[v]-k)*(n-m-d[v]+k))*edge[i].w;
					f[u][j]=max(f[u][j],f[u][j-k]+f[v][k]+sum);
				}
		}
	}
}
signed main()
{
	memset(head,-1,sizeof(head));	
	memset(f,-1,sizeof(f));
	Read(n,m);
	m=min(m,n-m);
	foru(i,1,n-1)
		Read(x,y,z),add(x,y,z),add(y,x,z);
	dfs(1,0);
	printf("%lld",f[1][m]);
}