H106OJ | 金属采集（Java）

写在前面

过不了！！！就很气！！！

题目

Description
人类在火星上发现了一种新的金属！这些金属分布在一些奇怪的地方，不妨叫它节点好了。一些节点之间有道路相连，所有的节点和道路形成了一棵树。一共有 n 个节点，这些节点被编号为 1~n 。人类将 k 个机器人送上了火星，目的是采集这些金属。这些机器人都被送到了一个指定的着落点， S 号节点。每个机器人在着落之后，必须沿着道路行走。当机器人到达一个节点时，它会采集这个节点蕴藏的所有金属矿。当机器人完成自己的任务之后，可以从任意一个节点返回地球。当然，回到地球的机器人就无法再到火星去了。我们已经提前测量出了每条道路的信息，包括它的两个端点 x 和 y，以及通过这条道路需要花费的能量 w 。我们想花费尽量少的能量采集所有节点的金属，这个任务就交给你了。
Input
第一行包含三个整数 n, S 和 k ，分别代表节点个数、着落点编号，和机器人个数。
接下来一共 n-1 行，每行描述一条道路。一行含有三个整数 x, y 和 w ，代表在 x 号节点和 y 号节点之间有一条道路，通过需要花费 w 个单位的能量。所有道路都可以双向通行。
Output
输出一个整数，代表采集所有节点的金属所需要的最少能量。
Sample Input 1
6 1 3
1 2 1
2 3 1
2 4 1000
2 5 1000
1 6 1000
Sample Output 1
3004
Hint
HINT:时间限制：1.0s 内存限制：256.0MB
所有机器人在 1 号节点着陆。
第一个机器人的行走路径为 1->6 ，在 6 号节点返回地球，花费能量为1000。
第二个机器人的行走路径为 1->2->3->2->4 ，在 4 号节点返回地球，花费能量为1003。
第一个机器人的行走路径为 1->2->5 ，在 5 号节点返回地球，花费能量为1001。

分析

是一道经典树形DP改变的题，从子节点推到根节点，基本框架一样，获取最小dp值的方式有些复杂；

准备

1. dp[num][robotnum]用来存储以某结点为子树的根节点机器人花费能量最小值；
2. 类node里面有某点的子节点，所需能量；
3. 容器list[num]某个节点的子节点数组（和？自己实现链表麻烦）；
4. visit[num]标记某点是否已被访问，这也是跳出循环的条件。

核心逻辑

递推公式：
参照下面代码；
封装一个dfs函数，从主函数获取的开始节点进入，遇到一个子节点就循环，一直访问到树叶才会跳出循环，回去执行循环下的代码：找当前节点的到若干子树的能量最小值，这就是和经典树形DP不同的地方，这不是一颗二叉树
，它有多个子树，而且必须访问所有节点，也就意味着必须从根节点走到所有子节点。

代码

import java.util.ArrayList;
import java.util.Scanner;

public class Main {
	static int[][] dp;
	static int[] visit;
	static ArrayList<Node>[] list;
	public static class Node{
		int son;
		int energy;
		Node(int s, int e) {
			son = s;
			energy = e;
		}
	}
	public static void dfs(int u) {
		int sz = list[u].size();
		visit[u] = 1;
		for (int i = 0; i <sz; i++) {
			int v = list[u].get(i).son;
			if (visit[v] == 1) continue;
			dfs(v);
			// 递推公式
			int sonEnergy = list[u].get(i).energy;
			for (int j = dp[i].length-1; j >= 0; j--) {
				dp[u][j] += dp[v][0] + 2 * sonEnergy;
				for (int k = 1; k <= j; k++) {
					dp[u][j] = Math.min(dp[u][j], dp[u][j-k]+k*sonEnergy + dp[v][k]);
				}
			}
		}
	}
	public static void main(String[] args) {
		Scanner input = new Scanner(System.in);
		int num = input.nextInt();
		int start = input.nextInt();
		int robotNum = input.nextInt();
		int[] p = new int[num+1];
		dp = new int[num+1][robotNum+1];
		visit = new int[num+1];
		list = new ArrayList[num+1];
		for (int i = 1; i < num; i++) {
			int a = input.nextInt();
			int b = input.nextInt();
			int e = input.nextInt();
			if (list[a] == null) 
				list[a] = new ArrayList();
			if (list[b] == null)
				list[b] = new ArrayList();
			list[a].add(new Node(b, e));
			list[b].add(new Node(a, e));
		}
		dfs(start);
		System.out.println(dp[start][robotNum]);
	}
}