【LeetCode：690】员工的重要性（Java）

题目链接

• 690. 员工的重要性

题目描述

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

给定一个员工数组 employees，其中：

employees[i].id 是第 i 个员工的 ID。
employees[i].importance 是第 i 个员工的重要度。
employees[i].subordinates 是第 i 名员工的直接下属的 ID 列表。
给定一个整数 id 表示一个员工的 ID，返回这个员工和他所有下属的重要度的 总和。

示例 1：

输入：employees = [[1,5,[2,3]],[2,3,[]],[3,3,[]]], id = 1
输出：11
解释：
员工 1 自身的重要度是 5 ，他有两个直系下属 2 和 3 ，而且 2 和 3 的重要度均为 3 。因此员工 1 的总重要度是 5 + 3 + 3 = 11 。

示例 2：

输入：employees = [[1,2,[5]],[5,-3,[]]], id = 5
输出：-3
解释：员工 5 的重要度为 -3 并且没有直接下属。
因此，员工 5 的总重要度为 -3。

提示：

1 <= employees.length <= 2000
1 <= employees[i].id <= 2000
所有的 employees[i].id 互不相同。
-100 <= employees[i].importance <= 100
一名员工最多有一名直接领导，并可能有多名下属。
employees[i].subordinates 中的 ID 都有效。

求解思路1

• 深搜（DFS）：通过哈希表将员工的id和员工进行映射，方便后续查找。搜索指定id员工的所有直接下属员工，并将其重要性值累加即可。

实现代码1

/*
// Definition for Employee.
class Employee {
    public int id;
    public int importance;
    public List<Integer> subordinates;
};
*/

class Solution {
    Map<Integer, Employee> map = new HashMap<>();

    public int getImportance(List<Employee> employees, int id) {
        for (Employee e : employees) {
            map.put(e.id, e);
        }
        return dfs(id);
    }

    public int dfs(int id) {
        Employee e = map.get(id);
        int sum = e.importance;
        List<Integer> subordinates = e.subordinates;
        // 搜索该id员工的所有直属员工
        for (int s : subordinates) {
            sum += dfs(s);
        }
        return sum;
    }
}

求解思路2

• 宽搜（BFS）：同样需要创建哈希表将员工id和员工进行映射。创建队列，每次遍历该员工出队后，将该员工的所有直接下属员工全部入队继续遍历，累加得到结果。

实现代码2

/*
// Definition for Employee.
class Employee {
    public int id;
    public int importance;
    public List<Integer> subordinates;
};
*/

class Solution {
    public int getImportance(List<Employee> employees, int id) {
        Map<Integer, Employee> map = new HashMap<>();
        for (Employee e : employees) {
            map.put(e.id, e);
        }
        int sum = 0;
        Queue<Integer> queue = new LinkedList<>();
        queue.offer(id);
        while (!queue.isEmpty()) {
            int curId = queue.poll();
            Employee employee = map.get(curId);
            sum += employee.importance;
            // 将出队员工的所有直属员工入队
            List<Integer> subordinates = employee.subordinates;
            for (int s : subordinates) {
                queue.offer(s);
            }
        }
        return sum;
    }
}