690. 员工的重要性-每日一题

一、题目

        给定一个保存员工信息的数据结构，它包含了员工 唯一的 id ，重要度 和 直系下属的 id 。

        比如，员工 1 是员工 2 的领导，员工 2 是员工 3 的领导。他们相应的重要度为 15 , 10 , 5 。那么员工 1 的数据结构是 [1, 15, [2]] ，员工 2的 数据结构是 [2, 10, [3]] ，员工 3 的数据结构是 [3, 5, []] 。注意虽然员工 3 也是员工 1 的一个下属，但是由于 并不是直系 下属，因此没有体现在员工 1 的数据结构中。

        现在输入一个公司的所有员工信息，以及单个员工 id ，返回这个员工和他所有下属的重要度之和。

点击查看原题

二、思路

        根据给出的单个员工id找到拥有该id的Employee对象，并查找到该Employee的下属，递归这个过程并求和即可。
        如果不采取记忆化，查找到一个给定id的Employee对象需要O(n)时间复杂度，使用HashMap进行优化，可以将查找单个Employee的时间复杂度降低到O(1)。

三、代码

class Solution {
    Map<Integer, Employee> employeeMap;
    public int getImportance(List<Employee> employees, int id) {
        employeeMap = new HashMap();
        for (Employee employee : employees) {
            employeeMap.put(employee.id, employee);	// 建立员工id和员工对象的映射关系
        }
        return findAndSum(employeeMap.get(id));
    }
    private int findAndSum(Employee employee) {
        if (employee == null) {
            return 0;
        }
        int ans = employee.importance;
        for (int subordinate : employee.subordinates) {
            ans += findAndSum(employeeMap.get(subordinate));	// 递归查找直系下属的重要性之和
        }
        return ans;
    }
}

        时间复杂度为O(n)，记忆了所有对象与id的映射关系，空间复杂度为O(n)。