蓝桥杯java算法学习（五）

注意：以下部分内容摘自Acwing，仅用于个人学习记录，不做任何商业用途。

DP合集(背包问题在上一章节)

（1）线性DP

（2）区间DP

（3）树形DP

例题：没有上司的舞会

题目描述：

Ural 大学有 N 名职员，编号为 1∼N1∼。

他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。

每个职员有一个快乐指数，用整数 Hi给出，其中 1≤i≤N1。

现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。

在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式

第一行一个整数 N。

接下来 N 行，第 i 行表示 i号职员的快乐指数 Hi。

接下来 N−1 行，每行输入一对整数 L,K，表示 K是 L的直接上司。（注意一下，后一个数是前一个数的父节点，不要搞反）。

输出格式

输出最大的快乐指数。

详解如下

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;


public class Main {
	static final int N=60010;
	static int [] e=new int[N];
	static int [] ne=new int[N];
	static int [] h=new int[N];
	static int [] w=new int[N];//存储快乐值
	static int [] st=new int[N];//标记是否有父亲结点
	static int f[][] = new int[N][2];//状态数组.即用0，1表示当前根节点是否选择
	static int n;
	static int idx=0;
	
	public static void add(int a,int b) {
		e[idx] = b;
		ne[idx] = h[a];
		h[a] = idx++;
	}
	public static void dfs(int u) {
		f[u][1] = w[u];//标记当前点选择的情况下，快乐度赋值
		for(int i = h[u]; i != -1; i = ne[i]){//遍历所有子树，计算选择和不选择情况下的快乐度数
			int j = e[i];
			dfs(j);//遍历当前子树，将子树中的情况都赋值快乐度。
			f[u][0] += Math.max(f[j][0],f[j][1]); //不选择当前节点
			f[u][1] += f[j][0]; //选择当前节点，为了使得快乐度最大，子树不应该参与
		}
	}
	
	public static void main(String[] args) throws IOException {
		BufferedReader br =new BufferedReader(new InputStreamReader(System.in));
		n=Integer.parseInt(br.readLine());
		Arrays.fill(h,-1);
		for(int i=1;i<=n;i++) {
			
			w[i]=Integer.parseInt(br.readLine());
			
		}
		for(int i=1;i<n;i++) {
			String s[] = br.readLine().split(" ");
	        int a = Integer.parseInt(s[0]);
	        int b = Integer.parseInt(s[1]);
			add(b,a);//因为b是上司因此需要把a连接到b的后面
			st[a]=1;
		}
		 int root = 1;
		    while(st[root]==1){//寻找父节点
		        root++;
		    }
		   dfs(root);
		   System.out.println(Math.max(f[root][0],f[root][1]));
	}
	

}