你有一个保存员工信息的数据结构,它包含了员工唯一的 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。
class Solution {
public int getImportance(List<Employee> employees, int id) {
HashMap<Integer, Employee> map = new HashMap<Integer, Employee>();
for(Employee e : employees){
map.put(e.id, e);
}
return dfs(map, id);
}
public int dfs(HashMap<Integer, Employee> map, int id){
Employee e = map.get(id);
int res = e.importance;
for(int subId : e.subordinates){
res += dfs(map, subId);
}
return res;
}
}
Comments NOTHING