一、题目

给定一个保存员工信息的数据结构，它包含了员工 唯一的 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)。

三、代码

  1. class Solution {
  2.     Map<Integer, Employee> employeeMap;
  3.     public int getImportance(List<Employee> employees, int id) {
  4.         employeeMap = new HashMap();
  5.         for (Employee employee : employees) {
  6.             employeeMap.put(employee.id, employee);    // 建立员工id和员工对象的映射关系
  7.         }
  8.         return findAndSum(employeeMap.get(id));
  9.     }
  10.     private int findAndSum(Employee employee) {
  11.         if (employee == null) {
  12.             return 0;
  13.         }
  14.         int ans = employee.importance;
  15.         for (int subordinate : employee.subordinates) {
  16.             ans += findAndSum(employeeMap.get(subordinate));    // 递归查找直系下属的重要性之和
  17.         }
  18.         return ans;
  19.     }
  20. }

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