工作室课题---金属采集（树形dp）

、、\Description

人类在火星上发现了一种新的金属！这些金属分布在一些奇怪的地方，不妨叫它节点好了。一些节点之间有道路相连，所有的节点和道路形成了一棵树。一共有 n 个节点，这些节点被编号为 1~n 。人类将 k 个机器人送上了火星，目的是采集这些金属。这些机器人都被送到了一个指定的着落点， S 号节点。每个机器人在着落之后，必须沿着道路行走。当机器人到达一个节点时，它会采集这个节点蕴藏的所有金属矿。当机器人完成自己的任务之后，可以从任意一个节点返回地球。当然，回到地球的机器人就无法再到火星去了。我们已经提前测量出了每条道路的信息，包括它的两个端点 x 和 y，以及通过这条道路需要花费的能量 w 。我们想花费尽量少的能量采集所有节点的金属，这个任务就交给你了。

Input

第一行包含三个整数 n, S 和 k ，分别代表节点个数、着落点编号，和机器人个数。

接下来一共 n-1 行，每行描述一条道路。一行含有三个整数 x, y 和 w ，代表在 x 号节点和 y 号节点之间有一条道路，通过需要花费 w 个单位的能量。所有道路都可以双向通行。

代码实现：

inline void init()
{
	memset(head,-1,sizeof(head));
	memset(dp,0,sizeof(dp));
	tot=0;
}
inline void add(int u,int v,int w)
{
	e[tot].next=head[u],e[tot].to=v,e[tot].w=w,head[u]=tot++;
}
inline void dfs(int u,int f)
{
	for(int i=head[u];i!=-1;i=e[i].next)
	{
		int to=e[i].to;
		if(to==f) continue;
		dfs(to,u);
		for(int r=k;r>=0;r--)
		{
			dp[u][r]+=dp[to][0]+2*e[i].w;//特殊情况
			for(int j=1;j<=r;j++) dp[u][r]=min(dp[u][r],dp[u][r-j]+dp[to][j]+j*e[i].w);
		}
	}
}

解题思路：dp[i][j]代表着以i为根的子树，利用j个机器人去收集这个子树的金属，收集完了就返回地球所需要的最小花费，那么最终输出dp[n][k]。
状态转移方程就是：dp[i][r]=min(dp[i][r],dp[i][r-j]+dp[to][j]+j*w).假设以i为根的子树中分配了r个机器人，就是分配给以to为根的子树的机器人中，没有一个返回了地球，以to为根的子树节点中的金属都被收集了，但是所以的机器人又返回了，收集的金属数目是不会随着机器人的数目变的，如果机器人数目多的话，代价反而大，因为i-to这一条边走的机器人多了，这种情况下，最优的就是派一个机器人下去收集，然后再返回。