题目描述
1.员工信息的定义如下:
public class Employee {
public int happy;
public List<Employee> nexts;
public Employee(int h) {
happy = h;
nexts = new ArrayList<>();
}
}
2.题目描述
公司的每个员工都符合 Employee 类的描述。整个公司的人员结构可以看作是一棵标准的、 没有环的多叉树 树的头节点是公司唯一的老板,除老板之外的每个员工都有唯一的直接上级 叶节点是没有任何下属的基层员工(subordinates列表为空),除基层员工外每个员工都有一个或多个直接下级 这个公司现在要办party,你可以决定哪些员工来,哪些员工不来,
规则:
1.如果某个员工来了,那么这个员工的所有直接下级都不能来
2.派对的整体快乐值是所有到场员工快乐值的累加
3.你的目标是让派对的整体快乐值尽量大 给定一棵多叉树的头节点boss,请返回派对的最大快乐值。
递归解法一.
解题思路:
递归算法就要从头节点考虑,第一个节点可来可不来,两种情况:
1.头节点来了,他的直接子节点,也是可来可不来,两种情况,分别去递归,找出最大的值.
2.头节点来了,.那么他的直接子节点都不能来,然后去递归,
下面我们代码去实现.
/**
递归过程 flag 为true 代表来,false 代表不来.
分别去递归
*/
public static int process(Employee cur, boolean flag) {
if (flag) { // 如果cur的上级来的话,cur没得选,只能不来
int ans = 0;
for (Employee next : cur.nexts) {
ans += process1(next, false);
}
return ans;
} else { // 如果cur的上级不来的话,cur可以选,可以来也可以不来
int p1 = cur.happy;
int p2 = 0;
for (Employee next : cur.nexts) {
p1 += process1(next, true);
p2 += process1(next, false);
}
return Math.max(p1, p2);
}
}
/**
* 主方法调用
*/
public static int maxHappy1(Employee emp) {
if (emp == null) {
return 0;
}
return process1(emp, false);
}
递归解法二
把过程分解掉,到每个节点上的来或者不来情况都去递归出结果,然后进行比较
/**
主方法调用
*/
public static int maxHappy2(Employee head) {
Info allInfo = process(head);
return Math.max(allInfo.no, allInfo.yes);
}
public static class Info {
public int no; //不来的情况
public int yes; //来的情况
public Info(int n, int y) {
no = n;
yes = y;
}
}
public static Info process(Employee x) {
if (x == null) {
return new Info(0, 0);
}
int no = 0; //不来的话 初始值为0
int yes = x.happy; //来的话初始值 是他本身的值
for (Employee next : x.nexts) {
//递归
Info nextInfo = process(next);
no += Math.max(nextInfo.no, nextInfo.yes);
yes += nextInfo.no;
}
return new Info(no, yes);
}