派对的最大快乐值

Posted 2020-12-22 laydown

import java.util.ArrayList;
import java.util.List;

/**
 * 派对的最大快乐值
 * <p>
 * 一棵多叉树代表员工的上下级关系，孩子节点是父节点的直接下级。
 * 节点代表员工，属性包括快乐值和孩子节点列表。
 * 大家参加了party，要求一个员工去了则它的所有直接下级都不能去，问参加party能得到的最大快乐值是多少。
 */
public class MaxHappy {

    public static int maxHappy(Employee boss) {
        if (boss == null) {
            return 0;
        }
        ResultInfo all = process(boss);
        return Math.max(all.yes, all.no);
    }

    public static ResultInfo process(Employee x) {
        if (x.nexts.isEmpty()) {
            return new ResultInfo(x.happy, 0);
        }
        int yes = x.happy;
        int no = 0;
        for (Employee next : x.nexts) {
            ResultInfo nextInfo = process(next);
            yes += nextInfo.no;
            no += Math.max(nextInfo.yes, nextInfo.no);
        }
        return new ResultInfo(yes, no);
    }

    /**
     * 返回的快乐值
     */
    public static class ResultInfo {

        // 参与的快乐值
        public int yes;

        // 不参与的快乐值
        public int no;

        public ResultInfo(int y, int n) {
            yes = y;
            no = n;
        }

    }

    /**
     * 员工
     */
    public static class Employee {

        // 快乐值
        public int happy;

        // 直接下级
        public List<Employee> nexts;

        public Employee(int h) {
            happy = h;
            nexts = new ArrayList<>();
        }

    }

}

/* 如有意见或建议，欢迎评论区留言；如发现代码有误，欢迎批评指正 */