To kill all the bugs is always easier than to capture their brains. A map is drawn for you, with all the rooms marked by the amount of bugs inside, and the possibility of containing a brain. The cavern's structure is like a tree in such a way that there is one unique path leading to each room from the entrance. To finish the battle as soon as possible, you do not want to wait for the troopers to clear a room before advancing to the next one, instead you have to leave some troopers at each room passed to fight all the bugs inside. The troopers never re-enter a room where they have visited before.
A starship trooper can fight against 20 bugs. Since you do not have enough troopers, you can only take some of the rooms and let the nerve gas do the rest of the job. At the mean time, you should maximize the possibility of capturing a brain. To simplify the problem, just maximize the sum of all the possibilities of containing brains for the taken rooms. Making such a plan is a difficult job. You need the help of a computer.
The last test case is followed by two -1's.
5 10 50 10 40 10 40 20 65 30 70 30 1 2 1 3 2 4 2 5 1 1 20 7 -1 -1
50 7
解题思路:
那么动态转移方程就出来了 dp【i】【j】 = max{dp【i】【j】,dp【i】【j-k】+dp【子树】【k】}
剩下的需要注意dfs 就跟递归似的 你想求此时的最优值就得求子树的最优值 求子树的最优值就得求子树的子树的最优值
然后在从子树向根进行dp更新 找出最优值就可以了
import java.util.ArrayList;
import java.util.Scanner;
public class Main {
private static int m;
private static int n;
private static Scanner scanner;
private static int[][] dp;
private static ArrayList<Integer>[] graph;
private static int[][] arr;
public static void main(String[] args) {
// TODO Auto-generated method stub
scanner = new Scanner(System.in);
while (true) {
n = scanner.nextInt();
m = scanner.nextInt();
if(n==-1&&m==-1)break;
createGraph();
if(m==0) {
System.out.println(0);
}else {
dp = new int [n+1][m+1];
dfs(1,0);
//从根节点开始搜
//最后直接输出根节点就可以了
//根节点此时是每层最优值的累加和
System.out.println(dp[1][m]);
}
}
}
private static void dfs(int r,int p) {//r号房间
//先进行初始化
int a1 = (arr[r][0]+19)/20;//a1是干掉r号房间需要几个士
for (int i = a1; i <=m; i++) {
dp[r][i] = arr[r][1];
}
for (int i = 0; i < graph[r].size(); i++) {
int h = graph[r].get(i);//获取子节点
if (h!=p) {
dfs(h,r);//求出子节点的最优值
for (int j = m; j >=a1; j--) {
for (int j2 = 1; j2 <= j-a1; j2++) {
dp[r][j] = Math.max(dp[r][j], dp[h][j2]+dp[r][j-j2]);
//核心方程 攻打r号房间有j个士兵时的最优值
//是 子节点由父节点给了多少个兵的最优值 加上攻打r号房间有j减去分出去兵的个数的最优值
}
}
}
}
}
private static void createGraph() {
arr = new int [n+1][2];
graph = new ArrayList [n+1];
for (int i = 1; i <=n; i++) {
arr[i][0] = scanner.nextInt();
arr[i][1] = scanner.nextInt();
}
for (int i = 0; i < graph.length; i++) {
graph[i] = new ArrayList<Integer>();
}
for (int i = 1; i < n; i++) {
int a = scanner.nextInt();
int b = scanner.nextInt();
graph[a].add(b);
graph[b].add(a);
}
}
}