注意:以下部分内容摘自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]));
}
}