题目链接:https://leetcode.cn/problems/employee-importance/ (opens new window)
难度:中等
你有一个保存员工信息的数据结构,它包含了员工唯一的 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
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
这道中等题也比较简单,给的是一个对象数组,既然给了数组最好就不要想着去根据层级关系去建一颗树了,因为太浪费时间和空间了。
直接遍历一遍数组记录下对象id与对象的映射关系,然后根据目标对象的id,去计算他和他下属的重要度总和。
具体实现为深度优先搜索,当然也可以用队列来做广度优先,还是比较简单的,不像是中等题。
/**
* Definition for Employee.
* function Employee(id, importance, subordinates) {
* this.id = id;
* this.importance = importance;
* this.subordinates = subordinates;
* }
*/
/**
* @param {Employee[]} employees
* @param {number} id
* @return {number}
*/
var GetImportance = function(employees, id) {
const ehash = new Map()
employees.forEach(e => {
ehash.set(e.id, e)
})
// 计算数组中所有员工的重要度(含下属)
const getSubImp = (ids) => {
let imp = 0
for(let i = 0; i < ids.length; i++) {
const emp = ehash.get(ids[i])
imp += emp.importance + getSubImp(emp.subordinates)
}
return imp
}
return getSubImp([id])
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31